From ece3ad13372146331e15cbaa6b574ee22ef87bf6 Mon Sep 17 00:00:00 2001 From: Kevin Date: Sat, 13 Aug 2022 22:21:40 -0400 Subject: [PATCH] fix: type hint --- 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 b03cc95283..a3153f1149 100644 --- a/src/sagemaker/spark/processing.py +++ b/src/sagemaker/spark/processing.py @@ -700,8 +700,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,