Use compat namespace to avoid adding import_tensorflow_extension to p…

…yarrow.* namespace Change-Id: I94bd60d98b64cab6105fbdda793bc82050dc640e
apache · Jul 4, 2018 · 92aef7a · 92aef7a
1 parent 2ca3de9
commit 92aef7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/pyarrow/__init__.py b/python/pyarrow/__init__.py
@@ -44,11 +44,11 @@ def parse_version(root):
         __version__ = None
 
 
-from pyarrow.compat import import_tensorflow_extension
+import pyarrow.compat as compat
 
 
 # Workaround for https://issues.apache.org/jira/browse/ARROW-2657
-import_tensorflow_extension()
+compat.import_tensorflow_extension()
 
 
 from pyarrow.lib import cpu_count, set_cpu_count