diff --git a/bluepyefe/recording.py b/bluepyefe/recording.py index e1be476..e86366d 100644 --- 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)