Fix for truth value of array

Author: kabilar

element_array_ephys/ephys_acute.py

@@ -551,7 +551,7 @@ def make(self, key):
                     'spike_times': unit_spike_times,
                     'spike_count': spike_count,
                     'spike_sites': spike_sites[kilosort_dataset.data['spike_clusters'] == unit],
-                    'spike_depths': spike_depths[kilosort_dataset.data['spike_clusters'] == unit] if spike_depths else None})
+                    'spike_depths': spike_depths[kilosort_dataset.data['spike_clusters'] == unit] if spike_depths.any() else None})
 
         self.insert1(key)
         self.Unit.insert([{**key, **u} for u in units])

element_array_ephys/ephys_chronic.py

@@ -550,7 +550,7 @@ def make(self, key):
                     'spike_times': unit_spike_times,
                     'spike_count': spike_count,
                     'spike_sites': spike_sites[kilosort_dataset.data['spike_clusters'] == unit],
-                    'spike_depths': spike_depths[kilosort_dataset.data['spike_clusters'] == unit] if spike_depths else None})
+                    'spike_depths': spike_depths[kilosort_dataset.data['spike_clusters'] == unit] if spike_depths.any() else None})
 
         self.insert1(key)
         self.Unit.insert([{**key, **u} for u in units])

element_array_ephys/ephys_precluster.py

@@ -685,7 +685,7 @@ def make(self, key):
                     'spike_times': unit_spike_times,
                     'spike_count': spike_count,
                     'spike_sites': spike_sites[kilosort_dataset.data['spike_clusters'] == unit],
-                    'spike_depths': spike_depths[kilosort_dataset.data['spike_clusters'] == unit] if spike_depths else None})
+                    'spike_depths': spike_depths[kilosort_dataset.data['spike_clusters'] == unit] if spike_depths.any() else None})
 
         self.insert1(key)
         self.Unit.insert([{**key, **u} for u in units])