Merge remote-tracking branch 'origin/benchmark-has-harms' into benchm…

…ark-has-harms
mlcommons · Jan 30, 2024 · 21d20f2 · 21d20f2
2 parents 0fed2a5 + 0558158
commit 21d20f2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_benchmark.py b/tests/test_benchmark.py
@@ -1,3 +1,4 @@
+import pathlib
 import pytest
 
 from coffee.benchmark import (
@@ -9,7 +10,9 @@
     quantize_stars,
 )
 from coffee.helm import HelmSut, BbqHelmTest, HelmResult
-from .test_helm_runner import SIMPLE_BBQ_DATA, SIMPLE_TOXICITY_DATA
+
+SIMPLE_BBQ_DATA = pathlib.Path(__file__).parent / "data/full_runs/simple_bbq"
+SIMPLE_TOXICITY_DATA = pathlib.Path(__file__).parent / "data/full_runs/toxicity"
 
 
 def test_benchmark_definition_basics():