Fix unit tests (#2034)
This commit is contained in:
@@ -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
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user