From 132c84aa4b96c97d1c9e9d6090dc15764ea59870 Mon Sep 17 00:00:00 2001 From: chrisiacovella Date: Thu, 14 Sep 2023 23:11:18 -0700 Subject: [PATCH] Working on updating unit tests. --- modelforge/tests/test_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modelforge/tests/test_dataset.py b/modelforge/tests/test_dataset.py index b565984d..de6650ff 100644 --- a/modelforge/tests/test_dataset.py +++ b/modelforge/tests/test_dataset.py @@ -182,7 +182,7 @@ def test_dataset_splitting(dataset): train_dataset, val_dataset, test_dataset = RandomSplittingStrategy().split(dataset) energy = train_dataset[0]["E"].item() - assert np.isclose(energy, -157.09958704371914) + assert np.isclose(energy, -412509.9375) print(energy) try: