[arrow] branch master updated: ARROW-3950: [Plasma] Make loading the TensorFlow op optional

[pc...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

pcmoritz pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/arrow.git


The following commit(s) were added to refs/heads/master by this push:
     new 4d810b4  ARROW-3950: [Plasma] Make loading the TensorFlow op optional
4d810b4 is described below

commit 4d810b4a9c37e79fde6b134ac90ee0c5f7f6c9bf
Author: Philipp Moritz <pc...@gmail.com>
AuthorDate: Fri Dec 7 18:27:27 2018 -0800

    ARROW-3950: [Plasma] Make loading the TensorFlow op optional
    
    Author: Philipp Moritz <pc...@gmail.com>
    
    Closes #3117 from pcmoritz/tf-optional-loading and squashes the following commits:
    
    0404e7ede <Philipp Moritz> fix
    2b0d25432 <Philipp Moritz> make loading the tensorflow op optional
---
 python/pyarrow/plasma.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/python/pyarrow/plasma.py b/python/pyarrow/plasma.py
index fbca9d5..056172c 100644
--- a/python/pyarrow/plasma.py
+++ b/python/pyarrow/plasma.py
@@ -39,7 +39,9 @@ TF_PLASMA_OP_PATH = os.path.join(pa.__path__[0], "tensorflow", "plasma_op.so")
 
 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)