fix _initialize_data_loaders in data_silo (#476)

deepset-ai · Aug 17, 2020 · a903289 · a903289
1 parent 153f4ed
commit a903289
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/farm/data_handler/data_silo.py b/farm/data_handler/data_silo.py
@@ -318,7 +318,7 @@ def _save_dataset_to_cache(self):
     def _initialize_data_loaders(self):
         """ Initializing train, dev and test data loaders for the already loaded datasets """
 
-        if self.processor.train_filename:
+        if self.data["train"] is not None:
             if self.distributed:
                 sampler_train = DistributedSampler(self.data["train"])
             else:
@@ -343,7 +343,7 @@ def _initialize_data_loaders(self):
         else:
             data_loader_dev = None
 
-        if self.processor.test_filename:
+        if self.data["test"] is not None:
             data_loader_test = NamedDataLoader(
                 dataset=self.data["test"],
                 sampler=SequentialSampler(self.data["test"]),