diff --git a/modules/analysis/rosbag-image-pipeline-sfn/tests/test_detect_scenes.py b/modules/analysis/rosbag-image-pipeline-sfn/tests/test_detect_scenes.py index 4f7599cf..dcc1c4ff 100644 --- a/modules/analysis/rosbag-image-pipeline-sfn/tests/test_detect_scenes.py +++ b/modules/analysis/rosbag-image-pipeline-sfn/tests/test_detect_scenes.py @@ -1,8 +1,9 @@ import os -from artifacts.detect_scenes import * from pyspark.sql.types import ArrayType, IntegerType +from artifacts.detect_scenes import * + def create_spark_session(port: int): os.environ["PYSPARK_SUBMIT_ARGS"] = '--packages "org.apache.hadoop:hadoop-aws:3.3.1" pyspark-shell'