Fixed bath

Hjorthmedh · Aug 21, 2024 · 2346bc8 · 2346bc8
1 parent 36bf912
commit 2346bc8
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/snudda/simulate/simulate.py b/snudda/simulate/simulate.py
@@ -2028,12 +2028,14 @@ def add_bath_application(self, species_name, concentration, time, neuron_id=None
 
         self.bath_application[species_name] = (time, concentration, t_vect, conc_vect)
 
-        for neuron in [self.neurons[x] for x in neuron_id]:
-            if neuron.modulation is not None:
-                neuron.modulation.concentration_from_vector(species_name=species_name,
-                                                            concentration_vector=conc_vect,
-                                                            time_vector=t_vect)
+        for nid in neuron_id:
+            if nid in self.neurons:
+                n = self.neurons[nid]
 
+                if n.modulation is not None:
+                    n.modulation.concentration_from_vector(species_name=species_name,
+                                                           concentration_vector=conc_vect,
+                                                           time_vector=t_vect)
 
     ############################################################################