diff --git a/src/modelgauge/tests/safe_v1.py b/src/modelgauge/tests/safe_v1.py index 7ad4e2e9..22624f97 100644 --- a/src/modelgauge/tests/safe_v1.py +++ b/src/modelgauge/tests/safe_v1.py @@ -191,7 +191,7 @@ class SafeTestVersion1(BaseSafeTestVersion1): def register_tests(cls, evaluator=None): - for locale in Locale: + for locale in [Locale.EN_US]: for hazard in cls.hazards: test_uid = BaseSafeTestVersion1.create_uid(hazard, locale, evaluator) if not test_uid in TESTS.keys(): diff --git a/tests/modelgauge_tests/test_safe.py b/tests/modelgauge_tests/test_safe.py index ccf40d6f..d800566a 100644 --- a/tests/modelgauge_tests/test_safe.py +++ b/tests/modelgauge_tests/test_safe.py @@ -321,7 +321,7 @@ def test_error_getting_undefined_annotators(self): def test_registered_tests(self): hazards = ["vcr", "ncr", "src", "cse", "dfm", "spc", "prv", "ipv", "iwp", "hte", "ssh", "sxc"] - locales = ["en_us", "fr_fr", "zh_cn", "hi_in"] + locales = ["en_us"] for hazard in hazards: for locale in locales: uid = f"safe-{hazard}-{locale}-1.0"