Working on updating unit tests.

choderalab · Sep 15, 2023 · 132c84a · 132c84a
1 parent 7b27580
commit 132c84a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: