FIX: fix bug of shuffle in data_loader

Author: iLampard

easy_tpp/preprocess/data_loader.py

@@ -105,11 +105,15 @@ def get_loader(self, split='train', **kwargs):
 
         dataset = TPPDataset(data)
         tokenizer = EventTokenizer(self.data_config.data_specs)
+
+        # Remove 'shuffle' from kwargs if it exists to avoid conflict
+        shuffle = kwargs.pop('shuffle', self.kwargs.get('shuffle', False))
+
         loader = get_data_loader(dataset,
                                  self.backend,
                                  tokenizer,
                                  batch_size=self.kwargs['batch_size'],
-                                 shuffle=self.kwargs['shuffle'],
+                                 shuffle=shuffle,
                                  **kwargs)
 
         return loader

version.py

@@ -1 +1 @@
-__version__ = '0.0.9'
+__version__ = '0.1.0'