Remove warning for synapse types

jaxleyverse · Nov 6, 2023 · 85c2a43 · 85c2a43
1 parent 4f9c52d
commit 85c2a43
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neurax/modules/network.py b/neurax/modules/network.py
@@ -263,11 +263,11 @@ def _step_synapse(
                 synapse_names[i] == type(synapse_type).__name__
             ), "Mixup in the ordering of synapses. Please create an issue on Github."
             synapse_states, synapse_current_terms = synapse_type.step(
-                u, delta_t, voltages, params, np.asarray(pre_syn_inds[i])
+                u, delta_t, voltages, params, np.asarray(pre_syn_inds[synapse_names[i]])
             )
             synapse_current_terms = postsyn_voltage_updates(
                 voltages,
-                np.asarray(post_syn_inds[i]),
+                np.asarray(post_syn_inds[synapse_names[i]]),
                 *synapse_current_terms,
             )
             syn_voltage_terms += synapse_current_terms[0]