Fix (#763)

mlcommons · Dec 18, 2024 · 1652675 · 1652675
1 parent fcdb319
commit 1652675
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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)