From a30e27c9ac14969e4b582dbc7b853c589ebac126 Mon Sep 17 00:00:00 2001 From: frostedoyster Date: Tue, 17 Dec 2024 10:03:59 +0100 Subject: [PATCH] Fix regression test --- .../experimental/nanopet/tests/test_regression.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/src/metatrain/experimental/nanopet/tests/test_regression.py b/src/metatrain/experimental/nanopet/tests/test_regression.py index bee99cf9..270ac927 100644 --- 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], ] )