Skip to content

Commit

Permalink
Add correct values for regression tests
Browse files Browse the repository at this point in the history
  • Loading branch information
frostedoyster committed Nov 13, 2024
1 parent 79e8b30 commit 8712d27
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions src/metatrain/experimental/nanopet/tests/test_regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -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],
]
)

Expand Down

0 comments on commit 8712d27

Please sign in to comment.