fix bad merge conflict

mlcommons · Dec 18, 2024 · de6d5a1 · de6d5a1
1 parent c826a38
commit de6d5a1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/modelbench/benchmarks.py b/src/modelbench/benchmarks.py
@@ -9,7 +9,7 @@
 from modelgauge.tests.safe_v1 import Locale
 
 from modelbench.hazards import HazardDefinition, HazardScore, SafeHazardV1, Standards, STANDARDS
-from modelbench.scoring import LetterGradeMixin, score_to_ordinal_grade, score_to_ordinal_grade_v1
+from modelbench.scoring import LetterGradeMixin, score_to_ordinal_grade
 from modelbench.uid import HasUid
 
 

diff --git a/tests/modelbench_tests/test_record.py b/tests/modelbench_tests/test_record.py
@@ -21,13 +21,13 @@
 
 
 @pytest.fixture()
-def benchmark_score(end_time):
+def benchmark_score(end_time, sut):
     bd = GeneralPurposeAiChatBenchmarkV1(Locale.EN_US, "practice")
     low_est = ValueEstimate.make(0.5, 10)
     high_est = ValueEstimate.make(0.8, 20)
     bs = BenchmarkScore(
         bd,
-        ModelGaugeSut.for_key("mistral-7b"),
+        sut,
         [
             HazardScore(
                 hazard_definition=SafeHazardV1("cse", Locale.EN_US, "practice"),