Ignore missing populations in make_h5types

GazzolaLab · Apr 6, 2024 · 6aaa347 · 6aaa347
1 parent 6204bec
commit 6aaa347
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/miv_simulator/utils/io.py b/src/miv_simulator/utils/io.py
@@ -274,7 +274,9 @@ def create_neural_h5(
 ) -> None:
     populations = []
     for pop_name, pop_idx in population_definitions.items():
-        layer_counts = cell_distributions.get(pop_name, {})
+        if pop_name not in cell_distributions:
+            continue
+        layer_counts = cell_distributions[pop_name]
         pop_count = 0
         for layer_name, layer_count in layer_counts.items():
             pop_count += layer_count