removed one order of magnitude precision required

tutorials/examples/test_scripts.py

@@ -69,13 +69,13 @@ def test_hypergrid(ndim: int, height: int):
     args = HypergridArgs(ndim=ndim, height=height, n_trajectories=n_trajectories)
     final_l1_dist = train_hypergrid_main(args)
     if ndim == 2 and height == 8:
-        assert np.isclose(final_l1_dist, 8.78e-4, atol=1e-5)
+        assert np.isclose(final_l1_dist, 8.78e-4, atol=1e-4)
     elif ndim == 2 and height == 16:
-        assert np.isclose(final_l1_dist, 4.56e-4, atol=1e-5)
+        assert np.isclose(final_l1_dist, 4.56e-4, atol=1e-4)
     elif ndim == 4 and height == 8:
-        assert np.isclose(final_l1_dist, 1.6e-4, atol=1e-5)
+        assert np.isclose(final_l1_dist, 1.6e-4, atol=1e-4)
     elif ndim == 4 and height == 16:
-        assert np.isclose(final_l1_dist, 2.45e-5, atol=1e-6)
+        assert np.isclose(final_l1_dist, 2.45e-5, atol=1e-5)
 
 
 @pytest.mark.parametrize("ndim", [2, 4])
@@ -85,13 +85,13 @@ def test_discreteebm(ndim: int, alpha: float):
     args = DiscreteEBMArgs(ndim=ndim, alpha=alpha, n_trajectories=n_trajectories)
     final_l1_dist = train_discreteebm_main(args)
     if ndim == 2 and alpha == 0.1:
-        assert np.isclose(final_l1_dist, 2.97e-3, atol=1e-3)
+        assert np.isclose(final_l1_dist, 2.97e-3, atol=1e-2)
     elif ndim == 2 and alpha == 1.0:
-        assert np.isclose(final_l1_dist, 0.017, atol=1e-3)
+        assert np.isclose(final_l1_dist, 0.017, atol=1e-2)
     elif ndim == 4 and alpha == 0.1:
-        assert np.isclose(final_l1_dist, 0.009, atol=1e-3)
+        assert np.isclose(final_l1_dist, 0.009, atol=1e-2)
     elif ndim == 4 and alpha == 1.0:
-        assert np.isclose(final_l1_dist, 0.062, atol=1e-3)
+        assert np.isclose(final_l1_dist, 0.062, atol=1e-2)
 
 
 @pytest.mark.parametrize("delta", [0.1, 0.25])
@@ -114,10 +114,10 @@ def test_box(delta: float, loss: str):
     print(args)
     final_jsd = train_box_main(args)
     if loss == "TB" and delta == 0.1:
-        assert np.isclose(final_jsd, 3.81e-2, atol=1e-3)
+        assert np.isclose(final_jsd, 3.81e-2, atol=1e-2)
     elif loss == "DB" and delta == 0.1:
-        assert np.isclose(final_jsd, 0.134, atol=1e-2)
+        assert np.isclose(final_jsd, 0.134, atol=1e-1)
     if loss == "TB" and delta == 0.25:
-        assert np.isclose(final_jsd, 0.0411, atol=1e-3)
+        assert np.isclose(final_jsd, 0.0411, atol=1e-2)
     elif loss == "DB" and delta == 0.25:
-        assert np.isclose(final_jsd, 0.0142, atol=1e-3)
+        assert np.isclose(final_jsd, 0.0142, atol=1e-2)