Fix charts

Author: jlubken

src/penn_chime/charts.py

@@ -35,16 +35,18 @@ def build_admits_chart(
             color=color,
             tooltip=tooltip,
         )
-        .configure_legend(orient="bottom")
-        .interactive()
     )
     bar = (
         alt.Chart()
         .encode(x=alt.X(**x))
         .transform_filter(alt.datum.day == 0)
         .mark_rule(color="black", opacity=0.35, size=2)
     )
-    return alt.layer(points, bar, data=admits_floor_df)
+    return (
+        alt.layer(points, bar, data=admits_floor_df)
+        .configure_legend(orient="bottom")
+        .interactive()
+    )
 
 
 def build_census_chart(
@@ -72,16 +74,18 @@ def build_census_chart(
             color=color,
             tooltip=tooltip,
         )
-        .configure_legend(orient="bottom")
-        .interactive()
     )
     bar = (
         alt.Chart()
         .encode(x=alt.X(**x))
         .transform_filter(alt.datum.day == 0)
         .mark_rule(color="black", opacity=0.35, size=2)
     )
-    return alt.layer(points, bar, data=census_floor_df)
+    return (
+        alt.layer(points, bar, data=census_floor_df)
+        .configure_legend(orient="bottom")
+        .interactive()
+    )
 
 
 def build_sim_sir_w_date_chart(
@@ -109,16 +113,18 @@ def build_sim_sir_w_date_chart(
             color=color,
             tooltip=tooltip,
         )
-        .configure_legend(orient="bottom")
-        .interactive()
     )
     bar = (
         alt.Chart()
         .encode(x=alt.X(**x))
         .transform_filter(alt.datum.day == 0)
         .mark_rule(color="black", opacity=0.35, size=2)
     )
-    return alt.layer(points, bar, data=sim_sir_w_date_floor_df)
+    return (
+        alt.layer(points, bar, data=sim_sir_w_date_floor_df)
+        .configure_legend(orient="bottom")
+        .interactive()
+    )
 
 
 def build_descriptions(