Add encoder to handle masked numbers.

theengineear · theengineear · commit 6d9e8ca32db0 · 2014-11-21T00:25:13.000-08:00
diff --git a/plotly/utils.py b/plotly/utils.py
@@ -126,6 +126,21 @@ def sageJSONEncoder(self, obj):
             pass
         return None
 
+    def maskedNumbersEncoder(self, obj):
+        """This catches masked numbers which can't be serialized.
+
+        Pandas (and others) may use masked numbers to signify data that's
+        not NaN, but also not valid in computations; something to be ignored.
+
+        """
+        import math
+        try:
+            if math.isnan(obj):
+                return float('NaN')
+        except:
+            pass
+        return None
+
     def builtinJSONEncoder(self, obj):
         try:
             return obj.to_plotly_json()
@@ -138,7 +153,7 @@ def default(self, obj):
         except TypeError as e:
             encoders = (self.builtinJSONEncoder, self.datetimeJSONEncoder,
                         self.numpyJSONEncoder, self.pandasJSONEncoder,
-                        self.sageJSONEncoder)
+                        self.sageJSONEncoder, self.maskedNumbersEncoder)
             for encoder in encoders:
                 s = encoder(obj)
                 if s is not None:
