Commit 19b93c93 authored by peastman's avatar peastman
Browse files

Fixed error

parent 4033af30
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -949,6 +949,7 @@ class DiskDataset(Dataset):
    self.tasks, self.metadata_df = self.load_metadata()
    self._cached_shards = None
    self._memory_cache_size = 20 * (1 << 20)  # 20 MB
    self._cache_used = 0

  @staticmethod
  def create_dataset(shard_generator, data_dir=None, tasks=[]):