redunce number of samples in dummy dataset of test to fix memory issu…

…es on conda-forge
torchmd · Jan 25, 2025 · eeca6ed · eeca6ed
1 parent 15afbcd
commit eeca6ed
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_datamodule.py b/tests/test_datamodule.py
@@ -61,6 +61,8 @@ def test_datamodule_standardize(energy, forces, has_atomref, tmpdir):
     args["val_size"] = 10
     args["test_size"] = 10
     args["log_dir"] = tmpdir
+    args["batch_size"] = 2
+    args["inference_batch_size"] = 2
 
     dataset = DummyDataset(
         num_samples=100, energy=energy, forces=forces, has_atomref=has_atomref