make figures have original widths

Author: mayofaulkner

ibllib/pipes/tasks.py

@@ -165,7 +165,7 @@ def register_images(self, **kwargs):
         if self.one and len(self.plot_tasks) > 0:
             for plot_task in self.plot_tasks:
                 try:
-                    _ = plot_task.register_images()
+                    _ = plot_task.register_images(widths=['orig'])
                 except Exception:
                     _logger.error(traceback.format_exc())
                     continue

ibllib/plots/figures.py

@@ -350,6 +350,22 @@ def get_probe_signature(self):
         output_signature = [('spike_sorting_raster*.png', f'snapshot/{self.pname}', True)]
         self.signature = {'input_files': input_signature, 'output_files': output_signature}
 
+    def get_signatures(self, **kwargs):
+        files_spikes = Path(self.session_path).joinpath('alf').rglob('spikes.times.npy')
+        folder_probes = [f.parent for f in files_spikes]
+
+        full_input_files = []
+        for sig in self.signature['input_files']:
+            for folder in folder_probes:
+                full_input_files.append((sig[0], str(folder.relative_to(self.session_path)), sig[2]))
+        if len(full_input_files) != 0:
+            self.input_files = full_input_files
+        else:
+            self.input_files = self.signature['input_files']
+
+        self.output_files = self.signature['output_files']
+
+
 
 class BadChannelsAp(ReportSnapshotProbe):
     """