Moved make_filename to BaseBatch

dataset/batch.py

@@ -79,14 +79,6 @@ def run_once(self, *args, **kwargs):
         _ = self.data, args, kwargs
         return [[]]
 
-    @staticmethod
-    def make_filename():
-        """ Generate unique filename for the batch """
-        random_data = np.random.uniform(0, 1, size=10) * 123456789
-        # probability of collision is around 2e-10.
-        filename = hexlify(random_data.data)[:8]
-        return filename.decode("utf-8")
-
     def infer_dtype(self, data=None):
         """ Detect dtype of batch data """
         if data is None:

dataset/batch_base.py

@@ -1,4 +1,5 @@
 """ Contains the base batch class """
+import numpy as np
 from .decorators import action
 
 
@@ -10,6 +11,14 @@ def __init__(self, index):
         self.index = index
         self._data = None
 
+    @staticmethod
+    def make_filename():
+        """ Generate unique filename for the batch """
+        random_data = np.random.uniform(0, 1, size=10) * 123456789
+        # probability of collision is around 2e-10.
+        filename = hexlify(random_data.data)[:8]
+        return filename.decode("utf-8")
+
     @action
     def load(self, src, fmt=None):
         """ Load data from a file or another data source """