ARROW-3950: [Plasma] Make loading the TensorFlow op optional

Author: Philipp Moritz <[email protected]> Closes apache#3117 from pcmoritz/tf-optional-loading and squashes the following commits: 0404e7e <Philipp Moritz> fix 2b0d254 <Philipp Moritz> make loading the tensorflow op optional
nyhu · Dec 8, 2018 · 4d810b4 · 4d810b4
1 parent 80989ef
commit 4d810b4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/pyarrow/plasma.py b/python/pyarrow/plasma.py
@@ -39,7 +39,9 @@
 
 tf_plasma_op = None
 
-if os.path.exists(TF_PLASMA_OP_PATH):
+
+def load_plasma_tensorflow_op():
+    global tf_plasma_op
     import tensorflow as tf
     tf_plasma_op = tf.load_op_library(TF_PLASMA_OP_PATH)