Skip to content

Commit e914f2a

Browse files
Merge branch 'notest' into develop
2 parents 91c6a8d + 11e9cbd commit e914f2a

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/penn_chime/charts.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
from altair import Chart # type: ignore
66
import pandas as pd # type: ignore
7+
import numpy as np
78

89
from .parameters import Parameters
910
from .utils import add_date_column
@@ -32,8 +33,15 @@ def new_admissions_chart(
3233
x_kwargs = {"shorthand": "day", "title": "Days from today"}
3334

3435
# TODO fix the fold to allow any number of dispositions
36+
37+
ceiled_admits = projection_admits.copy()
38+
39+
ceiled_admits.hospitalized = np.ceil(ceiled_admits.hospitalized)
40+
ceiled_admits.icu = np.ceil(ceiled_admits.icu)
41+
ceiled_admits.ventilated = np.ceil(ceiled_admits.ventilated)
42+
3543
return (
36-
alt.Chart(projection_admits.head(plot_projection_days))
44+
alt.Chart(ceiled_admits.head(plot_projection_days))
3745
.transform_fold(fold=["hospitalized", "icu", "ventilated"])
3846
.mark_line(point=True)
3947
.encode(

0 commit comments

Comments
 (0)