fix: use self.eq_param_index inside _diff_eqs_wrapper

NeuroDiffGym · Dec 31, 2023 · b90ab42 · b90ab42
1 parent b5d457d
commit b90ab42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neurodiffeq/solvers.py b/neurodiffeq/solvers.py
@@ -1357,7 +1357,7 @@ def __init__(self, ode_system, conditions, t_min, t_max,
 
         def _diff_eqs_wrapper(*variables):
             funcs_and_coords = variables[:N_FUNCTIONS + N_COORDS]
-            eq_params = tuple(variables[idx] for idx in eq_param_index)
+            eq_params = tuple(variables[idx] for idx in self.eq_param_index)
             return ode_system(*funcs_and_coords, *eq_params)
 
         super(BundleSolver1D, self).__init__(