handle pandas.NA

Author: nicolaskruchten

Diff for: packages/python/plotly/_plotly_utils/utils.py

@@ -167,15 +167,19 @@ def encode_as_sage(obj):
 
     @staticmethod
     def encode_as_pandas(obj):
-        """Attempt to convert pandas.NaT"""
+        """Attempt to convert pandas.NaT / pandas.NA"""
         pandas = get_module("pandas", should_load=False)
         if not pandas:
             raise NotEncodable
 
         if obj is pandas.NaT:
             return None
-        else:
-            raise NotEncodable
+
+        # pandas.NA was introduced in pandas 1.0
+        if hasattr(pandas, "NA") and obj is pandas.NA:
+            return None
+
+        raise NotEncodable
 
     @staticmethod
     def encode_as_numpy(obj):