Fix unit tests (#2034)

This commit is contained in:
Lianmin Zheng
2024-11-14 11:08:37 -08:00
committed by GitHub
parent c3eac1b010
commit aae5434bdf

View File

@@ -740,8 +740,7 @@ def run_mmlu_test(
try: try:
metrics = run_eval(args) metrics = run_eval(args)
print(f"{metrics=}") assert metrics["score"] >= 0.65, f"{metrics=}"
self.assertGreaterEqual(metrics["score"], 0.65)
finally: finally:
pass pass