update spark processing with kms keys=

Author: Chuyang Deng

src/sagemaker/spark/processing.py

@@ -174,6 +174,7 @@ def run(
         self,
         submit_app,
         inputs=None,
+        kms_key=None,
         outputs=None,
         arguments=None,
         wait=True,
@@ -188,6 +189,8 @@ def run(
             inputs (list[:class:`~sagemaker.processing.ProcessingInput`]): Input files for
                 the processing job. These must be provided as
                 :class:`~sagemaker.processing.ProcessingInput` objects (default: None).
+            kms_key (str): The ARN of the KMS key that is used to encrypt the
+                user code file (default: None).
             outputs (list[:class:`~sagemaker.processing.ProcessingOutput`]): Outputs for
                 the processing job. These can be specified as either path strings or
                 :class:`~sagemaker.processing.ProcessingOutput` objects (default: None).
@@ -207,6 +210,7 @@ def run(
         super().run(
             submit_app,
             inputs,
+            kms_key,
             outputs,
             arguments,
             wait,

tests/unit/sagemaker/spark/test_processing.py

@@ -200,7 +200,7 @@ 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)
+    mock_super_run.assert_called_with("app", None, None, None, None, True, True, None, None)
 
 
 @pytest.mark.parametrize(