change(tests): fix failing processing unit tests

Author: athewsey

tests/unit/sagemaker/spark/test_processing.py

@@ -200,7 +200,17 @@ def test_configuration_validation(config, expected, sagemaker_session) -> None:
 def test_spark_processor_base_run(mock_super_run, spark_processor_base):
     spark_processor_base.run(submit_app="app")
 
-    mock_super_run.assert_called_with("app", None, None, None, True, True, None, None, None)
+    mock_super_run.assert_called_with(
+        code="app",
+        inputs=None,
+        outputs=None,
+        arguments=None,
+        wait=True,
+        logs=True,
+        job_name=None,
+        experiment_config=None,
+        kms_key=None,
+    )
 
 
 @pytest.mark.parametrize(

tests/unit/sagemaker/workflow/test_steps.py

@@ -435,6 +435,7 @@ def test_processing_step_normalizes_args(mock_normalize_args, sagemaker_session)
         inputs=step.inputs,
         outputs=step.outputs,
         code=step.code,
+        source_dir=None,
     )