Skip to content

Commit 583d169

Browse files
committed
Update sir.py
ruff E741: fixed ambiguous variables names
1 parent 469104d commit 583d169

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

bayesflow/benchmarks/sir.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,10 +34,10 @@ def prior(rng: np.random.Generator = None):
3434
def _deriv(x, t, N, beta, gamma):
3535
"""Helper function for scipy.integrate.odeint."""
3636

37-
S, I, R = x
38-
dS = -beta * S * I / N
39-
dI = beta * S * I / N - gamma * I
40-
dR = gamma * I
37+
s, i, r = x
38+
dS = -beta * s * i / N
39+
dI = beta * s * i / N - gamma * i
40+
dR = gamma * i
4141
return dS, dI, dR
4242

4343

0 commit comments

Comments
 (0)