getSparkSession() instead of spark variable

Author: vikasgupta78

test/InMemPipeTestImages.py

@@ -10,7 +10,7 @@
 
 from PIL import Image
 
-df = (spark.read.json('/home/ubuntu/image_data/listings/metadata'))
+df = (getSparkSession().read.json('/home/ubuntu/image_data/listings/metadata'))
 
 df = (
   df
@@ -31,7 +31,7 @@
   )
 
 image_metadata = (
-  spark
+  getSparkSession()
     .read
     .csv(
       path='/home/ubuntu/image_data/images/metadata',

test/InMemPipeTestImagesJson.py

@@ -10,7 +10,7 @@
 
 from PIL import Image
 
-df = (spark.read.json('/home/ubuntu/image_data/listings/metadata'))
+df = (getSparkSession().read.json('/home/ubuntu/image_data/listings/metadata'))
 
 df = (
   df
@@ -31,7 +31,7 @@
   )
 
 image_metadata = (
-  spark
+  getSparkSession()
     .read
     .csv(
       path='/home/ubuntu/image_data/images/metadata',