Workaround for vega/vega-lite#6544

sacundim · May 26, 2020 · c7a1ac6 · c7a1ac6
1 parent 006e5e8
commit c7a1ac6
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/covid_19_puerto_rico/charts.py b/src/covid_19_puerto_rico/charts.py
@@ -307,7 +307,12 @@ def make_chart(self, df):
 
         heatmap = base.mark_rect().encode(
             color=alt.Color('value:Q', title=None, legend=None,
-                            scale=alt.Scale(scheme="redgrey", domainMid=0))
+                            scale=alt.Scale(scheme="redgrey", domainMid=0,
+                                            # WORKAROUND: Set the domain manually to forcibly
+                                            # include the domainMid or else we run into
+                                            # https://github.com/vega/vega-lite/issues/6544
+                                            domain=[min(0, df['value'].min()),
+                                                    max(0, df['value'].max())]))
         )
 
         text = base.mark_text(fontSize=9).encode(
@@ -359,7 +364,12 @@ def make_chart(self, df):
 
         heatmap = base.mark_rect().encode(
             color=alt.Color('value:Q', title=None, legend=None,
-                            scale=alt.Scale(scheme="redgrey", domainMid=0))
+                            scale=alt.Scale(scheme="redgrey", domainMid=0,
+                                            # WORKAROUND: Set the domain manually to forcibly
+                                            # include the domainMid or else we run into
+                                            # https://github.com/vega/vega-lite/issues/6544
+                                            domain=[min(0, df['value'].min()),
+                                                    max(0, df['value'].max())]))
         )
 
         text = base.mark_text(fontSize=4).encode(