Fix fixture names

nv-morpheus · Jan 12, 2024 · 6891868 · 6891868
1 parent 1347cb3
commit 6891868
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/dfencoder/test_scalers.py b/tests/dfencoder/test_scalers.py
@@ -33,14 +33,14 @@ def tensor_fixture():
     yield torch.tensor([7.4, 8.3, 9.5], dtype=torch.float32)
 
 
-@pytest.fixture(name="standard_scalar", scope="function")
+@pytest.fixture(name="standard_scaler", scope="function")
 def standard_scaler_fixture(fit_tensor):
     scaler = scalers.StandardScaler()
     scaler.fit(fit_tensor)
     yield scaler
 
 
-@pytest.fixture(name="modified_scalar", scope="function")
+@pytest.fixture(name="modified_scaler", scope="function")
 def modified_scaler_fixture(fit_tensor):
     scaler = scalers.ModifiedScaler()
     scaler.fit(fit_tensor)