diff --git a/datasets/cbm_dataframe_parser.py b/datasets/cbm_dataframe_parser.py index 56c42c43b..7991dbc2a 100644 --- a/datasets/cbm_dataframe_parser.py +++ b/datasets/cbm_dataframe_parser.py @@ -390,7 +390,7 @@ def __gen_datasets(self): np.ones([len(anomaly_features), 1])), axis=0) # Save pickle file pickle.dump((self.signal_list, self.lbl_list, self.speed_list, self.load_list), - open(self.dataset_pkl_file_path, 'wb')) + open(self.dataset_pkl_file_path, 'wb')) def __len__(self): if self.is_truncated: diff --git a/datasets/samplemotordatalimerick.py b/datasets/samplemotordatalimerick.py index 800540c31..6128b90fd 100755 --- a/datasets/samplemotordatalimerick.py +++ b/datasets/samplemotordatalimerick.py @@ -145,7 +145,7 @@ def parse_ADXL356C_and_return_common_df_row(self, file_full_path, sensor_sr_Hz, def __getitem__(self, index): if self.accel_in_second_dim and not self.speed_and_load_available: - signal, lbl = super().__getitem__(index) + signal, lbl = super().__getitem__(index) # pylint: disable=unbalanced-tuple-unpacking signal = torch.transpose(signal, 0, 1) lbl = lbl.transpose() return signal, lbl