Fix regression test

metatensor · Dec 17, 2024 · a30e27c · a30e27c
1 parent 11f1d7f
commit a30e27c
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/metatrain/experimental/nanopet/tests/test_regression.py b/src/metatrain/experimental/nanopet/tests/test_regression.py
@@ -53,8 +53,8 @@ def test_regression_init():
     )
 
     # if you need to change the hardcoded values:
-    torch.set_printoptions(precision=12)
-    print(output["mtt::U0"].block().values)
+    # torch.set_printoptions(precision=12)
+    # print(output["mtt::U0"].block().values)
 
     torch.testing.assert_close(
         output["mtt::U0"].block().values, expected_output, rtol=1e-5, atol=1e-5
@@ -115,11 +115,11 @@ def test_regression_train():
 
     expected_output = torch.tensor(
         [
-            [-0.016902115196],
-            [0.100093543530],
-            [0.038387011737],
-            [0.097679324448],
-            [0.118228666484],
+            [0.999483287334],
+            [0.587706387043],
+            [0.266740202904],
+            [0.535433053970],
+            [0.255621880293],
         ]
     )