Force evaluation for built-in map()

BlueBrain · Oct 9, 2023 · 35d16c9 · 35d16c9
1 parent b4aaba0
commit 35d16c9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/bluepyefe/plotting.py b/bluepyefe/plotting.py
@@ -116,6 +116,12 @@ def _plot(cell, output_dir, show=False):
 def plot_all_recordings(cells, output_dir, show=False, mapper=map):
     """Plot recordings for all cells and all protocols"""
     mapper(partial(_plot, output_dir=output_dir, show=show), cells)
+    if mapper == map:
+        # For the built-in map(), ensure immediate evaluation as it returns a lazy iterator
+        # which won't execute the function until iterated over. Converting to a list forces this iteration.
+        list(mapper(partial(_plot, output_dir=output_dir, show=show), cells))
+    else:
+        mapper(partial(_plot, output_dir=output_dir, show=show), cells)
 
 
 def plot_efeature(