update reproducability test

delphi-suite · Apr 27, 2024 · ab2fde2 · ab2fde2
1 parent ec4c75b
commit ab2fde2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/train/test_train_step.py b/tests/train/test_train_step.py
@@ -86,7 +86,7 @@ def test_basic_reproducibility(dataset, model):
 
     assert torch.isclose(
         params[[1000, 2000, 3000]],
-        torch.tensor([-0.01782517, -0.00771354, 0.03517739]),
+        torch.tensor([-0.01780166, -0.00762226, 0.03532362]),
     ).all()