fixed the destructor in Infinite

neuro-ml · Nov 1, 2023 · 186a0cd · 186a0cd
1 parent 6e03c9d
commit 186a0cd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dpipe/batch_iter/pipeline.py b/dpipe/batch_iter/pipeline.py
@@ -176,7 +176,8 @@ def __enter__(self):
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
-        if self.pipeline is not None:
+        # need getattr here, because the destructor might get called before the field is initialized
+        if getattr(self, 'pipeline', None) is not None:
             self.pipeline, pipeline = None, self.pipeline
             return pipeline.__exit__(exc_type, exc_val, exc_tb)