remove pool_trials arg

NeuralEnsemble · Dec 10, 2024 · e9e6778 · e9e6778
1 parent 303f363
commit e9e6778
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/elephant/statistics.py b/elephant/statistics.py
@@ -366,16 +366,12 @@ def _compute_fano(spiketrains: Union[List[neo.SpikeTrain], List[pq.Quantity], Li
             return spike_counts.var()/spike_counts.mean()
 
     if isinstance(spiketrains, elephant.trials.Trials):
-        if not pool_trials:
-            return [[_compute_fano([spiketrain]) for spiketrain in spiketrains.get_spiketrains_from_trial_as_list(idx)]
-                    for idx in range(spiketrains.n_trials)]
-        elif pool_trials:
-            list_of_lists_of_spiketrains = [
-                spiketrains.get_spiketrains_from_trial_as_list(trial_id=trial_no)
-                for trial_no in range(spiketrains.n_trials)]
-            return [_compute_fano([list_of_lists_of_spiketrains[trial_no][st_no]
-                                   for trial_no in range(len(list_of_lists_of_spiketrains))])
-                    for st_no in range(len(list_of_lists_of_spiketrains[0]))]
+        list_of_lists_of_spiketrains = [
+            spiketrains.get_spiketrains_from_trial_as_list(trial_id=trial_no)
+            for trial_no in range(spiketrains.n_trials)]
+        return [_compute_fano([list_of_lists_of_spiketrains[trial_no][st_no]
+                               for trial_no in range(len(list_of_lists_of_spiketrains))])
+                for st_no in range(len(list_of_lists_of_spiketrains[0]))]
     else:  # Legacy behavior
         return _compute_fano(spiketrains)