Merge pull request #79 from neuro-ml/develop

Develop
neuro-ml · Aug 1, 2023 · ab8f6bd · ab8f6bd
2 parents 21870d4 + e97e3f1
commit ab8f6bd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/dpipe/__version__.py b/dpipe/__version__.py
@@ -1 +1 @@
-__version__ = '0.2.6'
+__version__ = '0.2.7'
diff --git a/dpipe/batch_iter/pipeline.py b/dpipe/batch_iter/pipeline.py
@@ -146,7 +146,9 @@ def closing_callback(self):
         class ClosingCallback(Callback):
             def teardown(self, trainer, pl_module, stage):
                 this.close()
-
+            def on_exception(self, trainer, pl_module, exception):
+                this.close()
+
         this = self
         return ClosingCallback()