Fix flaky test_pp_single_node (#21564)

2026-03-27 14:33:46 -07:00
parent 38ad251738
commit 4a41aec844
1 changed files with 1 additions and 1 deletions
@@ -180,7 +180,7 @@ class TestQwenVLPPAccuracy(unittest.TestCase):
        metrics = run_eval_few_shot_gsm8k(args)
        print(f"{metrics=}")

-        self.assertGreater(metrics["accuracy"], 0.65)
+        self.assertGreaterEqual(metrics["accuracy"], 0.65)
        # Wait a little bit so that the memory check happens.
        time.sleep(4)