Update sir.py

ruff E741: fixed ambiguous variables names
bayesflow-org · Jul 3, 2024 · 583d169 · 583d169
1 parent 469104d
commit 583d169
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/bayesflow/benchmarks/sir.py b/bayesflow/benchmarks/sir.py
@@ -34,10 +34,10 @@ def prior(rng: np.random.Generator = None):
 def _deriv(x, t, N, beta, gamma):
     """Helper function for scipy.integrate.odeint."""
 
-    S, I, R = x
-    dS = -beta * S * I / N
-    dI = beta * S * I / N - gamma * I
-    dR = gamma * I
+    s, i, r = x
+    dS = -beta * s * i / N
+    dI = beta * s * i / N - gamma * i
+    dR = gamma * i
     return dS, dI, dR