Fix for EphysQC to look in local folders

Author: juhuntenburg

ibllib/ephys/ephysqc.py

@@ -42,9 +42,10 @@ class EphysQC(base.QC):
     default database if not given.
     """
 
-    def __init__(self, probe_id, **kwargs):
+    def __init__(self, probe_id, session_path=None, **kwargs):
         super().__init__(probe_id, endpoint='insertions', **kwargs)
         self.pid = probe_id
+        self.session_path = session_path
         self.stream = kwargs.pop('stream', True)
         keys = ('ap', 'ap_meta', 'lf', 'lf_meta')
         self.data = Bunch.fromkeys(keys)
@@ -57,7 +58,9 @@ def _ensure_required_data(self):
         """
         assert self.one is not None, 'ONE instance is required to ensure required data'
         eid, pname = self.one.pid2eid(self.pid)
-        self.probe_path = self.one.eid2path(eid).joinpath('raw_ephys_data', pname)
+        if self.session_path is None:
+            self.session_path = self.one.eid2path(eid)
+        self.probe_path = Path(self.session_path).joinpath('raw_ephys_data', pname)
         # Check if there is at least one meta file available
         meta_files = list(self.probe_path.rglob('*.meta'))
         assert len(meta_files) != 0, f'No meta files in {self.probe_path}'

ibllib/pipes/ephys_preprocessing.py

@@ -69,7 +69,7 @@ def _run(self, overwrite=False):
             pids = [p['id'] for p in create_alyx_probe_insertions(self.session_path, one=self.one)]
         qc_files = []
         for pid in pids:
-            eqc = ephysqc.EphysQC(pid, one=self.one)
+            eqc = ephysqc.EphysQC(pid, session_path=self.session_path, one=self.one)
             qc_files.extend(eqc.run(update=True, overwrite=overwrite))
         return qc_files