From 6aaa347c77d97ee578de21ac2e67316505ee37f1 Mon Sep 17 00:00:00 2001 From: Frithjof Gressmann Date: Sat, 6 Apr 2024 03:18:16 -0500 Subject: [PATCH] Ignore missing populations in make_h5types --- src/miv_simulator/utils/io.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/miv_simulator/utils/io.py b/src/miv_simulator/utils/io.py index 4f8301c..c00606c 100644 --- 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