test: reduce time diff

joschif · joschif · commit a5a4d1a87239 · 2025-08-05T14:35:26.000+02:00
diff --git a/tests/solver/test_solver.py b/tests/solver/test_solver.py
@@ -87,7 +87,7 @@ def test_predict_batch(self, dataloader, solver_class):
             atol=1e-1,
             rtol=1e-2,
         )
-        assert diff_nonbatched - diff_batched > 2
+        assert diff_nonbatched - diff_batched > 0.5
 
     @pytest.mark.parametrize("ema", [0.5, 1.0])
     def test_EMA(self, dataloader, ema):

Original file line number	Diff line number	Diff line change
`@@ -87,7 +87,7 @@ def test_predict_batch(self, dataloader, solver_class):`
`87`	`87`	`atol=1e-1,`
`88`	`88`	`rtol=1e-2,`
`89`	`89`	`)`
`90`		`- assert diff_nonbatched - diff_batched > 2`
	`90`	`+ assert diff_nonbatched - diff_batched > 0.5`
`91`	`91`
`92`	`92`	`@pytest.mark.parametrize("ema", [0.5, 1.0])`
`93`	`93`	`def test_EMA(self, dataloader, ema):`