slightly more robust

nicolaskruchten · nicolaskruchten · commit cf99af98eba7 · 2020-11-24T13:59:38.000-05:00
diff --git a/packages/python/plotly/plotly/figure_factory/_annotated_heatmap.py b/packages/python/plotly/plotly/figure_factory/_annotated_heatmap.py
@@ -196,14 +196,14 @@ def __init__(
             self.zmin = min([v for row in self.z for v in row])
             self.zmax = max([v for row in self.z for v in row])
 
-        if "zmin" in kwargs:
+        if kwargs.get("zmin", None) is not None:
             self.zmin = kwargs["zmin"]
-        if "zmax" in kwargs:
+        if kwargs.get("zmax", None) is not None:
             self.zmax = kwargs["zmax"]
 
         self.zmid = (self.zmax + self.zmin) / 2
 
-        if "zmid" in kwargs:
+        if kwargs.get("zmid", None) is not None:
             self.zmin = kwargs["zmid"]
 
     def get_text_color(self):
