fix: remove added trainer arg

IntelLabs · Oct 4, 2024 · ce112c6 · ce112c6
1 parent 4c34741
commit ce112c6
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/matsciml/models/tests/test_tasks.py b/matsciml/models/tests/test_tasks.py
@@ -84,9 +84,7 @@ def test_force_regression(egnn_config):
         },
     )
     task = ForceRegressionTask(**egnn_config)
-    trainer = pl.Trainer(
-        num_sanity_val_steps=0, max_steps=5, logger=False, enable_checkpointing=False
-    )
+    trainer = pl.Trainer(max_steps=5, logger=False, enable_checkpointing=False)
     trainer.fit(task, datamodule=devset)
     # make sure losses are tracked
     for key in ["energy", "force"]:
@@ -131,9 +129,7 @@ def test_force_regression_with_stress(egnn_config):
         },
     )
     task = ForceRegressionTask(**egnn_config)
-    trainer = pl.Trainer(
-        num_sanity_val_steps=0, max_steps=5, logger=False, enable_checkpointing=False
-    )
+    trainer = pl.Trainer(max_steps=5, logger=False, enable_checkpointing=False)
     trainer.fit(task, datamodule=devset)
     # make sure losses are tracked
     for key in ["energy", "force"]: