Skip to content

Commit 2736c0b

Browse files
committed
fix flake8
1 parent b97ff89 commit 2736c0b

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

tests/unit/sagemaker/spark/test_processing.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -742,14 +742,11 @@ def test_py_spark_processor_run(
742742
@patch("sagemaker.spark.processing._SparkProcessorBase._stage_submit_deps")
743743
@patch("sagemaker.spark.processing._SparkProcessorBase._generate_current_job_name")
744744
def test_py_spark_processor_run_twice(
745-
mock_generate_current_job_name,
746-
mock_stage_submit_deps,
747-
mock_super_run,
748-
py_spark_processor
745+
mock_generate_current_job_name, mock_stage_submit_deps, mock_super_run, py_spark_processor
749746
):
750747
mock_stage_submit_deps.return_value = (processing_input, "opt")
751748
mock_generate_current_job_name.return_value = "jobName"
752-
expected_command = ['smspark-submit', '--py-files', 'opt', '--jars', 'opt', '--files', 'opt']
749+
expected_command = ["smspark-submit", "--py-files", "opt", "--jars", "opt", "--files", "opt"]
753750
py_spark_processor.run(
754751
submit_app="submit_app",
755752
submit_py_files="files",
@@ -768,6 +765,7 @@ def test_py_spark_processor_run_twice(
768765

769766
assert py_spark_processor.command == expected_command
770767

768+
771769
@pytest.mark.parametrize(
772770
"config, expected",
773771
[

0 commit comments

Comments
 (0)