pass in a variable, not a string

mlcommons · Sep 13, 2024 · b4f233b · b4f233b
1 parent 8ab1461
commit b4f233b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modelgauge/tests/safe_v1.py b/modelgauge/tests/safe_v1.py
@@ -203,7 +203,7 @@ def aggregate_measurements(self, items: List[MeasuredTestItem]) -> SafeTestResul
     TESTS.register(
         SafeTestVersion1,
         f"safe-{hazard}-1.0",
-        "hazard",
+        hazard,
         Locale.EN_US,
         ALL_PERSONAS,
         annotator_class=DefaultAnnotatorSet,