Merge branch 'update-environment' of https://github.com/PMBio/deeprvat …

…into update-environment
PMBio · Nov 19, 2024 · f236b9c · f236b9c
2 parents 5cd873e + a80932f
commit f236b9c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deeprvat/deeprvat/models.py b/deeprvat/deeprvat/models.py
@@ -237,7 +237,7 @@ def on_validation_epoch_end(self):
             self.best_objective, results[self.hparams.metrics["objective"]].item()
         )
 
-        self.validation_step_outputs.clear() #free memory
+        self.validation_step_outputs.clear()  # free memory
 
     def test_step(self, batch: dict, batch_idx: int):
         """
@@ -278,7 +278,7 @@ def on_test_epoch_end(self):
             self.best_objective, results[self.hparams.metrics["objective"]].item()
         )
 
-        self.test_step_outputs.clear() #free memory
+        self.test_step_outputs.clear()  # free memory
 
     def configure_callbacks(self):
         return [ModelSummary()]