fix minor bug

mathLab · Feb 4, 2025 · f6a57ef · f6a57ef
1 parent 5d189cc
commit f6a57ef
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pina/data/data_module.py b/pina/data/data_module.py
@@ -265,6 +265,7 @@ def _apply_shuffle(condition_dict, len_data):
                 dataset_dict[key].update({condition_name: data})
         return dataset_dict
 
+
     def _create_dataloader(self, split, dataset):
         shuffle = self.shuffle if split == 'train' else False
         # Use custom batching (good if batch size is large)
@@ -313,7 +314,7 @@ def test_dataloader(self):
         """
         Create the testing dataloader
         """
-        raise self._create_dataloader('test', self.test_dataset)
+        return self._create_dataloader('test', self.test_dataset)
 
     def predict_dataloader(self):
         """