Merge pull request #345 from yahoo/leewyang_tb_path

Add sys.path to TensorBoard search path

Author: leewyang

requirements.txt

@@ -1,4 +1,4 @@
 py4j
-pyspark
+pyspark==2.3.1
 sphinx
 tensorflow

tensorflowonspark/TFSparkNode.py

@@ -207,7 +207,9 @@ def _mapfn(iter):
       # search for tensorboard in python/bin, PATH, and PYTHONPATH
       pypath = sys.executable
       pydir = os.path.dirname(pypath)
-      search_path = os.pathsep.join([pydir, os.environ['PATH'], os.environ['PYTHONPATH']])
+      sys_path = os.pathsep.join(sys.path)
+      search_path = os.pathsep.join([pydir, sys_path, os.environ['PATH'], os.environ['PYTHONPATH']])
+
       tb_path = util.find_in_path(search_path, 'tensorboard')                             # executable in PATH
       if not tb_path:
         tb_path = util.find_in_path(search_path, 'tensorboard/main.py')                   # TF 1.3+