From 8712d279542f72b9065a2975618f60b64a7624de Mon Sep 17 00:00:00 2001 From: frostedoyster Date: Wed, 13 Nov 2024 09:29:21 +0100 Subject: [PATCH] Add correct values for regression tests --- .../experimental/nanopet/tests/test_regression.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/metatrain/experimental/nanopet/tests/test_regression.py b/src/metatrain/experimental/nanopet/tests/test_regression.py index be3acadd..7b7cbd7b 100644 --- a/src/metatrain/experimental/nanopet/tests/test_regression.py +++ b/src/metatrain/experimental/nanopet/tests/test_regression.py @@ -44,11 +44,11 @@ def test_regression_init(): expected_output = torch.tensor( [ - [0.053602740169], - [-0.142421141267], - [-0.122775360942], - [-0.202754884958], - [-0.031394608319], + [-3.663903951645], + [-1.964396238327], + [-0.490164071321], + [-1.506513595581], + [-0.514931380749], ] )