fix

apache · Jul 27, 2018 · 5f0cafa · 5f0cafa
1 parent 2751679
commit 5f0cafa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/pyarrow/compat.py b/python/pyarrow/compat.py
@@ -243,6 +243,7 @@ def import_pytorch_extension():
     https://issues.apache.org/jira/browse/ARROW-2920
     """
     import ctypes
+    import os
 
     for path in _iterate_python_module_paths("torch"):
         ctypes.CDLL(os.path.join(path, "lib/libcaffe2.so"))