#0: Change assert

tenstorrent · Dec 10, 2024 · c460dc3 · c460dc3
1 parent f05f253
commit c460dc3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/models/demos/llama3/tests/test_llama_accuracy.py b/models/demos/llama3/tests/test_llama_accuracy.py
@@ -390,5 +390,9 @@ def test_tt_model_accuracy(
     )
 
     logger.info(f"Top-1: {total_top1_acc:.0f}% | Top-5: {total_top5_acc:.0f}%")
-    assert total_top1_acc > min_top1_acc, f"Top-1 accuracy {total_top1_acc:.1f}% is too low (expected >{min_top1_acc}%)"
-    assert total_top5_acc > min_top5_acc, f"Top-5 accuracy {total_top5_acc:.1f}% is too low (expected >{min_top5_acc}%)"
+    assert (
+        total_top1_acc >= min_top1_acc
+    ), f"Top-1 accuracy {total_top1_acc:.1f}% is too low (expected >={min_top1_acc}%)"
+    assert (
+        total_top5_acc >= min_top5_acc
+    ), f"Top-5 accuracy {total_top5_acc:.1f}% is too low (expected >={min_top5_acc}%)"