diff --git a/tests/model_tests/conftest.py b/tests/model_tests/conftest.py index 502f3e685..ab71a02cd 100644 --- a/tests/model_tests/conftest.py +++ b/tests/model_tests/conftest.py @@ -22,7 +22,7 @@ def tiny_model_paths(request, run_level): if not hasattr(item, "callspec"): continue model_name = item.callspec.params.get("model_name") - if model_name is None: + if model_name is None or model_name not in DIFFUSION_TEST_SETTINGS: continue if model_name not in model_paths: settings = DIFFUSION_TEST_SETTINGS[model_name]