calculate the time of testing

ai4os · Dec 19, 2024 · cb6425f · cb6425f
1 parent 75f1a9f
commit cb6425f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_predictions/conftest.py b/tests/test_predictions/conftest.py
@@ -98,7 +98,7 @@ def input_files(model_name):
     return options
 
 
-@pytest.fixture(scope="module", params=[model_names[0]])
+@pytest.fixture(scope="module", params=[model_names[1],model_names[2]])
 def model_name(request):
     """Fixture to provide the model_name argument to api.predict."""