updated SparkSession in test files

bigdatagenomics · Jan 12, 2018 · 7365b28 · 7365b28
1 parent e25a12a
commit 7365b28
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/adam-python/bdgenomics/adam/test/__init__.py b/adam-python/bdgenomics/adam/test/__init__.py
@@ -22,8 +22,6 @@
 import tempfile
 import unittest
 
-
-from pyspark.context import SparkContext
 from pyspark.sql import SparkSession
 
 class SparkTestCase(unittest.TestCase):

diff --git a/scripts/jenkins-test-pyadam.py b/scripts/jenkins-test-pyadam.py
@@ -18,10 +18,10 @@
 
 from bdgenomics.adam.adamContext import ADAMContext
 
-from pyspark.context import SparkContext
+from pyspark.sql import SparkSession
 
-sc = SparkContext('local')
-ac = ADAMContext(sc)
+ss = SparkSession.builder.master('local').getOrCreate()
+ac = ADAMContext(ss)
 
 reads = ac.loadAlignments("adam-core/src/test/resources/small.sam").toDF().count()