Skip to content

Commit 6b7f0c5

Browse files
authored
fix: fix flaky spark processor integ (#5109)
* fix: fix flaky spark processor integ * format
1 parent 9d8e1f5 commit 6b7f0c5

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

tests/integ/test_spark_processing.py

+4-8
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
SPARK_PATH = os.path.join(DATA_DIR, "spark")
3636

3737

38-
@pytest.fixture(scope="module")
38+
@pytest.fixture(scope="module", autouse=True)
3939
def build_jar():
4040
jar_file_path = os.path.join(SPARK_PATH, "code", "java", "hello-java-spark")
4141
# compile java file
@@ -207,12 +207,10 @@ def configuration() -> list:
207207

208208

209209
def test_sagemaker_pyspark_v3(
210-
spark_v3_py_processor, spark_v3_jar_processor, sagemaker_session, configuration, build_jar
210+
spark_v3_py_processor, spark_v3_jar_processor, sagemaker_session, configuration
211211
):
212212
test_sagemaker_pyspark_multinode(spark_v3_py_processor, sagemaker_session, configuration)
213-
test_sagemaker_java_jar_multinode(
214-
spark_v3_jar_processor, sagemaker_session, configuration, build_jar
215-
)
213+
test_sagemaker_java_jar_multinode(spark_v3_jar_processor, sagemaker_session, configuration)
216214

217215

218216
def test_sagemaker_pyspark_multinode(spark_py_processor, sagemaker_session, configuration):
@@ -280,9 +278,7 @@ def test_sagemaker_pyspark_multinode(spark_py_processor, sagemaker_session, conf
280278
assert len(output_contents) != 0
281279

282280

283-
def test_sagemaker_java_jar_multinode(
284-
spark_jar_processor, sagemaker_session, configuration, build_jar
285-
):
281+
def test_sagemaker_java_jar_multinode(spark_jar_processor, sagemaker_session, configuration):
286282
"""Test SparkJarProcessor using Java application jar"""
287283
bucket = spark_jar_processor.sagemaker_session.default_bucket()
288284
with open(os.path.join(SPARK_PATH, "files", "data.jsonl")) as data:

0 commit comments

Comments
 (0)