fix test

Author: Lee Yang

scripts/travis_before_install.sh

@@ -20,10 +20,6 @@ export PATH=$SPARK_HOME/bin:$PATH
 export JAVA_HOME=/usr/lib/jvm/java-8-openjdk-amd64
 
 # Update Python
+# Note: TensorFlow 2.0.0 requires pip>19.0
 pip install --upgrade pip
 export PYTHONPATH=$(pwd)
-
-# Echo versions
-java -version
-pyspark --version
-python --version

test/test_pipeline.py

@@ -141,7 +141,7 @@ def rdd_generator():
       tf_feed.terminate()
 
     # create a Spark DataFrame of training examples (features, labels)
-    rdd = self.sc.parallelize(self.train_examples, 3)
+    rdd = self.sc.parallelize(self.train_examples, 2)
     trainDF = rdd.toDF(['col1', 'col2'])
 
     # train and export model