Skip to content

Commit 3d69ca8

Browse files
committed
Reduce assignment to self.raw
1 parent 55677d5 commit 3d69ca8

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

src/penn_chime/models.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ def gen_policy(self, p: Parameters) -> Sequence[Tuple[float, int]]:
193193
]
194194

195195
def run_projection(self, p: Parameters, policy: Sequence[Tuple[float, int]]):
196-
self.raw = sim_sir(
196+
raw = sim_sir(
197197
self.susceptible,
198198
self.infected,
199199
p.recovered,
@@ -202,11 +202,12 @@ def run_projection(self, p: Parameters, policy: Sequence[Tuple[float, int]]):
202202
policy
203203
)
204204

205-
calculate_dispositions(self.raw, self.rates, p.market_share)
206-
calculate_admits(self.raw, self.rates)
207-
calculate_census(self.raw, self.days)
205+
calculate_dispositions(raw, self.rates, p.market_share)
206+
calculate_admits(raw, self.rates)
207+
calculate_census(raw, self.days)
208208

209-
self.current_infected = self.raw["infected"][self.i_day]
209+
self.current_infected = raw["infected"][self.i_day]
210+
self.raw = raw
210211

211212
def get_loss(self) -> float:
212213
"""Squared error: predicted vs. actual current hospitalized."""

0 commit comments

Comments
 (0)