Skip to content

Commit 4dd472c

Browse files
committed
fix: fix flaky spark processor integ
1 parent d0ccacf commit 4dd472c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/integ/test_spark_processing.py

Lines changed: 4 additions & 4 deletions
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,11 +207,11 @@ 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)
213213
test_sagemaker_java_jar_multinode(
214-
spark_v3_jar_processor, sagemaker_session, configuration, build_jar
214+
spark_v3_jar_processor, sagemaker_session, configuration
215215
)
216216

217217

@@ -281,7 +281,7 @@ def test_sagemaker_pyspark_multinode(spark_py_processor, sagemaker_session, conf
281281

282282

283283
def test_sagemaker_java_jar_multinode(
284-
spark_jar_processor, sagemaker_session, configuration, build_jar
284+
spark_jar_processor, sagemaker_session, configuration
285285
):
286286
"""Test SparkJarProcessor using Java application jar"""
287287
bucket = spark_jar_processor.sagemaker_session.default_bucket()

0 commit comments

Comments
 (0)