Fix census forecast

Compute census forecasts as far forward as admissions

Author: jlubken

src/penn_chime/charts.py

@@ -45,7 +45,7 @@ def build_admits_chart(
         .transform_filter(alt.datum.day == 0)
         .mark_rule(color="black", opacity=0.35, size=2)
     )
-    return alt.layer(points, bar, data=admits_df)
+    return alt.layer(points, bar, data=ceil_df)
 
 
 

src/penn_chime/models.py

@@ -312,7 +312,7 @@ def build_census_df(
         'date': admits_df.date,
         **{
             key: (
-                admits_df[key].cumsum().iloc[:-los]
+                admits_df[key].cumsum()
                 - admits_df[key].cumsum().shift(los).fillna(0)
             ).apply(np.ceil)
             for key, los in lengths_of_stay.items()

tests/test_app.py

@@ -245,7 +245,7 @@ def test_admits_chart():
     admits_df = pd.read_csv("tests/by_doubling_time/2020-03-28_projected_admits.csv")
     chart = build_admits_chart(alt=alt, admits_df=admits_df)
     assert isinstance(chart, (alt.Chart, alt.LayerChart))
-    assert round(chart.data.iloc[40].icu, 0) == 38
+    assert round(chart.data.iloc[40].icu, 0) == 39
 
     # test fx call with no params
     with pytest.raises(TypeError):