From 310b351df8dc91b5f5fca6ae02e3e912b85c2826 Mon Sep 17 00:00:00 2001 From: Caleb Robinson Date: Mon, 13 Jun 2022 09:34:30 -0700 Subject: [PATCH] Update conftest.py --- tests/trainers/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/trainers/conftest.py b/tests/trainers/conftest.py index 287c03b868a..3da741c3013 100644 --- 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: