Skip to content

Commit 83acfd1

Browse files
committed
Fix current_date
1 parent cb52025 commit 83acfd1

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/penn_chime/parameters.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def __init__(
5757
relative_contact_rate: float,
5858
mitigation_date: Optional[date] = None,
5959
ventilated: Disposition,
60-
current_date: date = date.today(),
60+
current_date: Optional[date] = None,
6161
date_first_hospitalized: Optional[date] = None,
6262
doubling_time: Optional[float] = None,
6363
infectious_days: int = 14,
@@ -69,7 +69,6 @@ def __init__(
6969
region: Optional[Regions] = None,
7070
):
7171
self.current_hospitalized = Positive(value=current_hospitalized)
72-
7372
Rate(value=hospitalized.rate), Rate(value=icu.rate), Rate(value=ventilated.rate)
7473
StrictlyPositive(value=hospitalized.days), StrictlyPositive(value=icu.days),
7574
StrictlyPositive(value=ventilated.days)
@@ -87,8 +86,10 @@ def __init__(
8786
else:
8887
raise AssertionError('population or regions must be provided.')
8988

89+
if current_date is None:
90+
current_date = date.today()
9091
self.current_date = Date(value=current_date)
91-
92+
9293
self.date_first_hospitalized = OptionalDate(value=date_first_hospitalized)
9394
self.doubling_time = OptionalStrictlyPositive(value=doubling_time)
9495

0 commit comments

Comments
 (0)