From 32af4087250df80b8c114fd73dde7a230ea4cb79 Mon Sep 17 00:00:00 2001 From: Steven Anton Date: Thu, 13 Oct 2022 15:55:15 -0700 Subject: [PATCH] Fix type annotations --- src/sagemaker/spark/processing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sagemaker/spark/processing.py b/src/sagemaker/spark/processing.py index 6f27df98db..04d55afbf0 100644 --- a/src/sagemaker/spark/processing.py +++ b/src/sagemaker/spark/processing.py @@ -680,8 +680,8 @@ class PySparkProcessor(_SparkProcessorBase): def __init__( self, role: str, - instance_type: Union[int, PipelineVariable], - instance_count: Union[str, PipelineVariable], + instance_type: Union[str, PipelineVariable], + instance_count: Union[int, PipelineVariable], framework_version: Optional[str] = None, py_version: Optional[str] = None, container_version: Optional[str] = None,