ARROW-4269: [Python] Fix serialization in pandas 0.22

Author: Philipp Moritz <[email protected]> Closes apache#3410 from pcmoritz/fix-pandas-0.22 and squashes the following commits: 54f13bb <Philipp Moritz> update dc6ffbd <Philipp Moritz> update 67133d0 <Philipp Moritz> make change consistent b6cba56 <Philipp Moritz> fix serialization in pandas 0.22
yupbank · Jan 16, 2019 · fbfd1ff · fbfd1ff
1 parent 143558e
commit fbfd1ff
Showing 1 changed file with 21 additions and 20 deletions.
diff --git a/python/pyarrow/serialization.py b/python/pyarrow/serialization.py
@@ -174,26 +174,27 @@ def _deserialize_pandas_series(data):
         custom_serializer=_pickle_to_buffer,
         custom_deserializer=_load_pickle_from_buffer)
 
-    if hasattr(pd.core.arrays, 'interval'):
-        context.register_type(
-            pd.core.arrays.interval.IntervalArray,
-            'pd.core.arrays.interval.IntervalArray',
-            custom_serializer=_pickle_to_buffer,
-            custom_deserializer=_load_pickle_from_buffer)
-
-    if hasattr(pd.core.arrays, 'period'):
-        context.register_type(
-            pd.core.arrays.period.PeriodArray,
-            'pd.core.arrays.period.PeriodArray',
-            custom_serializer=_pickle_to_buffer,
-            custom_deserializer=_load_pickle_from_buffer)
-
-    if hasattr(pd.core.arrays, 'datetimes'):
-        context.register_type(
-            pd.core.arrays.datetimes.DatetimeArray,
-            'pd.core.arrays.datetimes.DatetimeArray',
-            custom_serializer=_pickle_to_buffer,
-            custom_deserializer=_load_pickle_from_buffer)
+    if hasattr(pd.core, 'arrays'):
+        if hasattr(pd.core.arrays, 'interval'):
+            context.register_type(
+                pd.core.arrays.interval.IntervalArray,
+                'pd.core.arrays.interval.IntervalArray',
+                custom_serializer=_pickle_to_buffer,
+                custom_deserializer=_load_pickle_from_buffer)
+
+        if hasattr(pd.core.arrays, 'period'):
+            context.register_type(
+                pd.core.arrays.period.PeriodArray,
+                'pd.core.arrays.period.PeriodArray',
+                custom_serializer=_pickle_to_buffer,
+                custom_deserializer=_load_pickle_from_buffer)
+
+        if hasattr(pd.core.arrays, 'datetimes'):
+            context.register_type(
+                pd.core.arrays.datetimes.DatetimeArray,
+                'pd.core.arrays.datetimes.DatetimeArray',
+                custom_serializer=_pickle_to_buffer,
+                custom_deserializer=_load_pickle_from_buffer)
 
     context.register_type(
         pd.DataFrame, 'pd.DataFrame',