Fix

mlcommons · Dec 18, 2024 · 6960c45 · 6960c45
1 parent fcdb319
commit 6960c45
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)