diff --git a/src/modelgauge/tests/safe_v1.py b/src/modelgauge/tests/safe_v1.py index b334c27b..d71080c8 100644 --- a/src/modelgauge/tests/safe_v1.py +++ b/src/modelgauge/tests/safe_v1.py @@ -247,8 +247,7 @@ class SafeTestVersion1(BaseSafeTestVersion1): def register_tests(cls, evaluator=None): - for _locale in Locale: - locale = _locale.value + for locale in Locale: for hazard in cls.hazards: for prompt_set in PROMPT_SETS: test_uid = BaseSafeTestVersion1.create_uid(hazard, locale, prompt_set, evaluator)