Linter fix

asyatrhl · Mar 5, 2024 · 5084e69 · 5084e69
1 parent 1225e3a
commit 5084e69
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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