Fix BaseImagesBatch.load error on path creation with src=None

analysiscenter · Nov 7, 2022 · 61c6bb9 · 61c6bb9
1 parent a2cd076
commit 61c6bb9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/batchflow/batch.py b/batchflow/batch.py
@@ -90,7 +90,7 @@ class Batch(metaclass=MethodsTransformingMeta):
 
     def __init__(self, index, dataset=None, pipeline=None, preloaded=None, copy=False, *args, **kwargs):
         _ = args
-        if  self.components is not None and not isinstance(self.components, tuple):
+        if self.components is not None and not isinstance(self.components, tuple):
             raise TypeError("components should be a tuple of strings with components names")
         self.index = index
         self._preloaded_lock = threading.Lock()

diff --git a/batchflow/batch_image.py b/batchflow/batch_image.py
@@ -49,8 +49,9 @@ def _make_path(self, ix, src=None):
         path : str
             Full path to an element.
         """
-
-        if isinstance(src, FilesIndex):
+        if src is None:
+            path = str(ix)
+        elif isinstance(src, FilesIndex):
             path = src.get_fullpath(ix)
         elif isinstance(self.index, FilesIndex):
             path = self.index.get_fullpath(ix)