fix shuffle bug

Author: iLampard

README.md

@@ -108,6 +108,7 @@ Per users' request, we processed two non-anthropogenic datasets
 Explore the following tutorials that can be opened directly in Google Colab:
 
 - [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/ant-research/EasyTemporalPointProcess/blob/main/notebooks/easytpp_1_dataset.ipynb) Tutorial 1: Dataset in EasyTPP.
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/ant-research/EasyTemporalPointProcess/blob/main/notebooks/easytpp_2_tfb_wb.ipynb) Tutorial 2: Tensorboard in EasyTPP.
 
 ### End-to-end Example
 

easy_tpp/config_factory/runner_config.py

@@ -26,7 +26,6 @@ def __init__(self, base_config, model_config, data_config, trainer_config):
         self.base_config = base_config
         self.trainer_config = trainer_config
 
-        self.ensure_valid_config()
         self.update_config()
 
         # save the complete config
@@ -87,15 +86,6 @@ def parse_from_yaml_config(yaml_config, **kwargs):
             trainer_config=trainer_config
         )
 
-    def ensure_valid_config(self):
-        """Do some sanity check about the config, to avoid conflicts in settings.
-        """
-
-        # during testing we dont do shuffle by default
-        self.trainer_config.shuffle = False
-
-        return
-
     def update_config(self):
         """Updated config dict.
         """

easy_tpp/preprocess/data_loader.py

@@ -136,6 +136,8 @@ def test_loader(self, **kwargs):
         Returns:
             EasyTPP.DataLoader: data loader for test set.
         """
+        # for test set, we do not shuffle
+        kwargs['shuffle'] = False
         return self.get_loader('test', **kwargs)
 
     def get_statistics(self, split='train'):