Adding 0.05 tolerance

Author: lotif

tests/integration/models/clavaddpm/test_model.py

@@ -285,7 +285,10 @@ def test_train_single_table(tmp_path: Path):
     # expected_model_layers = list(expected_model_data.keys())
     # if np.allclose(model_data[model_layers[0]].detach(), expected_model_data[expected_model_layers[0]].detach()):
     # if the first layer is equal with minimal tolerance, all others should be equal as well
-    assert all(np.allclose(model_data[layer].detach(), expected_model_data[layer].detach()) for layer in model_layers)
+    assert all(
+        np.allclose(model_data[layer].detach(), expected_model_data[layer].detach(), atol=0.05)
+        for layer in model_layers
+    )
 
     # TODO: Figure out if there is a good way of testing the synthetic data results
     # on multiple platforms. https://app.clickup.com/t/868f43wp0
@@ -348,7 +351,10 @@ def test_train_multi_table(tmp_path: Path):
 
     # if np.allclose(model_data[model_layers[0]].detach(), expected_model_data[expected_model_layers[0]].detach()):
     # if the first layer is equal with minimal tolerance, all others should be equal as well
-    assert all(np.allclose(model_data[layer].detach(), expected_model_data[layer].detach()) for layer in model_layers)
+    assert all(
+        np.allclose(model_data[layer].detach(), expected_model_data[layer].detach(), atol=0.05)
+        for layer in model_layers
+    )
 
     #     # TODO: Figure out if there is a good way of testing the synthetic data results
     #     # on multiple platforms. https://app.clickup.com/t/868f43wp0