Remove hard coded section locations in network_builder

jonescompneurolab · Jul 18, 2022 · 9db8532 · 9db8532
1 parent c081dd2
commit 9db8532
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/hnn_core/network_builder.py b/hnn_core/network_builder.py
@@ -495,9 +495,11 @@ def _connect_celltypes(self):
 
                     # get synapse locations
                     syn_keys = list()
-                    if loc in ['proximal', 'distal']:
+                    # Targeting group of sections like proximal or distal
+                    if loc in target_cell.sect_loc:
                         for sect in target_cell.sect_loc[loc]:
                             syn_keys.append(f'{sect}_{receptor}')
+                    # Targeting individual section like soma or apical_tuft
                     else:
                         syn_keys = [f'{loc}_{receptor}']