Refactor dataset cache

mila-iqia · Aug 28, 2017 · d923152 · d923152
1 parent 756d20c
commit d923152
Show file tree

Hide file tree

Showing 2 changed files with 197 additions and 207 deletions.
diff --git a/fuel/datasets/hdf5.py b/fuel/datasets/hdf5.py
@@ -11,7 +11,7 @@
 from fuel.datasets import Dataset
 from fuel.schemes import SequentialExampleScheme
 from fuel.utils import do_not_pickle_attributes, Subset
-from fuel.utils.cache import dataset_cache
+from fuel.utils.cache import cache_file
 
 
 @do_not_pickle_attributes('nodes', 'h5file')
@@ -174,7 +174,7 @@ def __init__(self, file_or_path, which_sets, subset=None,
             self.external_file_handle = file_or_path
         else:
             self.path = file_or_path
-            self.path = dataset_cache.cache_file(self.path)
+            self.path = cache_file(self.path)
             self.external_file_handle = None
         which_sets_invalid_value = (
             isinstance(which_sets, six.string_types) or