fix behavior of BlankDataset

delira-dev · Feb 7, 2019 · 353648e · 353648e
1 parent beec333
commit 353648e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/delira/data_loading/dataset.py b/delira/data_loading/dataset.py
@@ -138,7 +138,7 @@ def get_subset(self, indices):
                     continue
                 kwargs[key] = val
 
-        kwargs["old_getitem"] = self.__getitem__
+        kwargs["old_getitem"] = self.__class__.__getitem__
         subset_data = [self.get_sample_from_index(idx) for idx in indices]
 
         return BlankDataset(subset_data, **kwargs)
@@ -220,7 +220,7 @@ def __getitem__(self, index):
             dictionary containing a single sample
 
         """
-        return self._old_getitem(index)
+        return self._old_getitem(self, index)
 
     def __len__(self):
         """