修复test_mql事件后积分步长控制

PythonModels/core/solver.py

@@ -10,8 +10,9 @@ class SolveIVPConfig:
     t_stop: float = 20.0
     method: str = "BDF"
     rtol: float = 1e-6
-    atol: float = 1e-8
+    atol: float = 1e-10
     max_step: float = 1e-3
+    first_step: float | None = None
 
 
 @dataclass(frozen=True)
@@ -91,12 +92,16 @@ def integrate_ode(
     except ImportError:
         return _runge_kutta_4(rhs, initial_state, config, t_eval)
 
-    return solve_ivp(
-        fun=rhs,
-        t_span=(config.t_start, config.t_stop),
-        y0=initial_state,
-        method=config.method,
-        rtol=config.rtol,
-        atol=config.atol,
-        t_eval=t_eval,
-    )
+    solve_options = {
+        "fun": rhs,
+        "t_span": (config.t_start, config.t_stop),
+        "y0": initial_state,
+        "method": config.method,
+        "rtol": config.rtol,
+        "atol": config.atol,
+        "max_step": config.max_step,
+        "t_eval": t_eval,
+    }
+    if config.first_step is not None:
+        solve_options["first_step"] = config.first_step
+    return solve_ivp(**solve_options)