flake

Author: themattinthehatt

brainbox/io/one.py

@@ -1572,7 +1572,8 @@ def load_pose(self, likelihood_thr=0.9, views=['left', 'right', 'body'], tracker
         tracker = 'lightningPose' if tracker in ['lp', 'litpose'] else tracker
         self.pose = {}
         for view in views:
-            pose_raw = self.one.load_object(self.eid, f'{view}Camera', attribute=[tracker, 'times'], revision=self.revision or None)
+            pose_raw = self.one.load_object(
+                self.eid, f'{view}Camera', attribute=[tracker, 'times'], revision=self.revision or None)
             # Double check if video timestamps are correct length or can be fixed
             times_fixed, dlc = self._check_video_timestamps(view, pose_raw['times'], pose_raw[tracker])
             self.pose[f'{view}Camera'] = likelihood_threshold(dlc, likelihood_thr)

ibllib/oneibl/patcher.py

@@ -682,7 +682,7 @@ def patch_dataset(self, file_list, dry=False, ftp=False, force=False, **kwargs):
             _logger.error(f'Files: {", ".join([f.name for f in file_list])} already exist, to overwrite set force=True')
             return
 
-        response = super().patch_dataset(file_list, dry=dry, repository=self.s3_repo, ftp=False, force=force,  **kwargs)
+        response = super().patch_dataset(file_list, dry=dry, repository=self.s3_repo, ftp=False, force=force, **kwargs)
         # TODO in an ideal case the flatiron filerecord won't be altered when we register this dataset. This requires
         # changing the the alyx.data.register_view
         for ds in response:

ibllib/pipes/video_tasks.py

@@ -828,7 +828,7 @@ def signature(self):
             # the following are required for the LP plot only
             # they are not strictly required, some plots just might be skipped
             # In particular the raw videos don't need to be downloaded as they can be streamed
-                           # [(f'_iblrig_{cam}Camera.raw.mp4', self.device_collection, True) for cam in self.cameras] +
+            # [(f'_iblrig_{cam}Camera.raw.mp4', self.device_collection, True) for cam in self.cameras] +
                            [(f'{cam}ROIMotionEnergy.position.npy', 'alf', False) for cam in self.cameras] +
                            [(f'{cam}Camera.ROIMotionEnergy.npy', 'alf', False) for cam in self.cameras] +
             # The trials table is used in the LP QC, however this is not an essential dataset

ibllib/tests/qc/test_dlc_qc.py

@@ -106,8 +106,7 @@ def test_check_lick_detection(self):
         outcome = self.qc.check_lick_detection()
         self.assertEqual('NOT_SET', outcome)
         self.qc.side = 'left'
-        self.qc.data['pose_coords'] = {'tongue_end_l': np.ones((2, 10)),
-                                      'tongue_end_r': np.ones((2, 10))}
+        self.qc.data['pose_coords'] = {'tongue_end_l': np.ones((2, 10)), 'tongue_end_r': np.ones((2, 10))}
         outcome = self.qc.check_lick_detection()
         self.assertEqual('FAIL', outcome)
         self.qc.data['pose_coords']['tongue_end_l'] *= np.nan