diff --git a/tests/modelbench_tests/test_record.py b/tests/modelbench_tests/test_record.py index f9a03d35..5ff867dd 100644 --- a/tests/modelbench_tests/test_record.py +++ b/tests/modelbench_tests/test_record.py @@ -78,7 +78,7 @@ def test_v1_hazard_definition_without_tests_loaded(): def test_v1_hazard_definition_with_tests_loaded(): hazard = SafeHazardV1("dfm", Locale.EN_US, "practice") - hazard.tests({"together": {"api_key": "ignored"}, "modellab_files": {"token": "ignored"}}) + hazard.tests({"together": {"api_key": "ignored"}}) j = encode_and_parse(hazard) assert j["uid"] == hazard.uid assert j["tests"] == ["safe-dfm-en_us-practice-1.0"]