add _sources

neuronsimulator · Jan 28, 2024 · cc8bc56 · cc8bc56
1 parent f3b093f
commit cc8bc56
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/share/lib/python/neuron/rxd/node.py b/share/lib/python/neuron/rxd/node.py
@@ -10,6 +10,7 @@
 
 from collections.abc import Callable
 
+_sources = []
 
 # function to change extracellular diffusion
 set_diffusion = nrn_dll_sym("set_diffusion")
@@ -349,9 +350,10 @@ def include_flux(self, *args, **kwargs):
                     f = float(args[0])
                     source = f
                     success = True
-                except:
+                except ValueError:
                     arith = rxdmath._ensure_arithmeticed(args[0])
                     source = lambda: eval_arith_flux(arith, self.region, self)
+                    _sources.append(source)
                     scale = 1 / self.volume
                     success = True
             if not success: