Update the testing pipeline to fix potential memory overflow issues when testing on large-scale datasets.

Author: yisongfu

basicts/runners/base_tsf_runner.py

@@ -519,7 +519,7 @@ def on_test_end(self, save_metrics: bool = False):
             self.logger.info(f'Test results saved to {os.path.join(self.ckpt_save_dir, "test_results")}.')
         if save_metrics:
             self.logger.info(f'Test metrics saved to {os.path.join(self.ckpt_save_dir, "test_metrics.json")}.')
-    
+
     @master_only
     def _save_test_results(self, batch_idx: int, batch_data: Dict[str, np.ndarray]) -> None:
 
@@ -536,7 +536,7 @@ def _save_test_results(self, batch_idx: int, batch_data: Dict[str, np.ndarray])
         """
 
         total_samples = len(self.test_data_loader.dataset)
-        
+
         save_dir = os.path.join(self.ckpt_save_dir, 'test_results')
         os.makedirs(save_dir, exist_ok=True)
         inputs_path = os.path.join(save_dir, 'inputs.npy')