instantiate one if not parsed as argument

Yiman00 · Nov 5, 2020 · 3d0a378 · 3d0a378
1 parent 4ac01d8
commit 3d0a378
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/brainbox/io/one.py b/brainbox/io/one.py
@@ -345,6 +345,7 @@ def load_spike_sorting_with_channel(eid, one=None, probe=None, dataset_types=Non
     :return: spikes, clusters, channels (dict of bunch, 1 bunch per probe)
     """
     # --- Get spikes and clusters data
+    one = one or ONE()
     dic_spk_bunch, dic_clus = load_spike_sorting(eid, one=one, probe=probe,
                                                  dataset_types=dataset_types, force=force)
     # -- Get brain regions and assign to clusters