Update conftest.py

microsoft · Jun 13, 2022 · 310b351 · 310b351
1 parent 3c8ee4c
commit 310b351
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/trainers/conftest.py b/tests/trainers/conftest.py
@@ -25,7 +25,7 @@ def state_dict(model: Module) -> Dict[str, Tensor]:
     return model.state_dict()
 
 
-@pytest.fixture(params=["classification_model", "encoder"])
+@pytest.fixture(params=["classification_model", "encoder_name"])
 def checkpoint(
     state_dict: Dict[str, Tensor], request: SubRequest, tmp_path: Path
 ) -> str: