fix imports

apache · Aug 15, 2017 · 7069e20 · 7069e20
1 parent 2171761
commit 7069e20
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/python/pyarrow/serialization.pxi b/python/pyarrow/serialization.pxi
@@ -20,7 +20,13 @@ from libcpp.vector cimport vector as c_vector
 from cpython.ref cimport PyObject
 from cython.operator cimport dereference as deref
 
-import cloudpickle as pickle
+try:
+    import cloudpickle as pickle
+except ImportError:
+    try:
+        import cPickle as pickle
+    except ImportError:
+        import pickle
 
 from pyarrow.lib cimport Buffer, NativeFile, check_status, _RecordBatchFileWriter
 

diff --git a/python/pyarrow/tests/test_serialization.py b/python/pyarrow/tests/test_serialization.py
@@ -24,7 +24,6 @@
 import string
 import sys
 
-from deepeq import deep_eq
 import pyarrow as pa
 import numpy as np