handle None values in compute_efeatures (#177)

BlueBrain · Apr 23, 2024 · 4bfe32a · 4bfe32a
1 parent b5fc0e5
commit 4bfe32a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/bluepyefe/recording.py b/bluepyefe/recording.py
@@ -312,8 +312,11 @@ def compute_efeatures(
         efel_vals = self.call_efel(efeatures, efel_settings)
         for efeature_name, efeature in zip(efeature_names, efeatures):
 
-            value = efel_vals[0][efeature]
-            if value is None or len(value) == 0 or numpy.isinf(numpy.nanmean(value)):
+            if efel_vals[0][efeature] is not None:
+                value = [v for v in efel_vals[0][efeature] if v is not None]
+            else:
+                value = []
+            if len(value) == 0 or numpy.isinf(numpy.nanmean(value)):
                 self.efeatures[efeature_name] = numpy.nan
             else:
                 self.efeatures[efeature_name] = numpy.nanmean(value)