Fix regression tests

metatensor · Dec 1, 2023 · e5d69e7 · e5d69e7
1 parent 6faddcb
commit e5d69e7
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/src/metatensor_models/soap_bpnn/tests/test_regression.py b/src/metatensor_models/soap_bpnn/tests/test_regression.py
@@ -31,11 +31,11 @@ def test_regression_init():
     )
     expected_output = torch.tensor(
         [
-            [0.2789987370],
-            [0.2335722791],
-            [0.0116647061],
-            [0.1048521983],
-            [0.0591454534],
+            [0.354014973507],
+            [0.079493871143],
+            [0.111979416192],
+            [0.405708668342],
+            [0.130853761777],
         ],
         dtype=torch.float64,
     )
@@ -65,11 +65,11 @@ def test_regression_train():
     output = soap_bpnn(structures[:5])
     expected_output = torch.tensor(
         [
-            [-0.9295628263],
-            [-0.8175271284],
-            [-0.5669038926],
-            [-0.4979601703],
-            [-0.5088090318],
+            [-0.928599079860],
+            [-0.825878482412],
+            [-0.474396235869],
+            [-0.463211805738],
+            [-0.514200923850],
         ],
         dtype=torch.float64,
     )