Skip to content

Commit fd61bf9

Browse files
ivan-khvostishkovJoseJuan98
authored andcommitted
fix: Fixing type hints for Spark processor that has instance type/count params in reverse order (aws#3444)
1 parent 7dce07f commit fd61bf9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/spark/processing.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -960,8 +960,8 @@ class SparkJarProcessor(_SparkProcessorBase):
960960
def __init__(
961961
self,
962962
role: str,
963-
instance_type: Union[int, PipelineVariable],
964-
instance_count: Union[str, PipelineVariable],
963+
instance_type: Union[str, PipelineVariable],
964+
instance_count: Union[int, PipelineVariable],
965965
framework_version: Optional[str] = None,
966966
py_version: Optional[str] = None,
967967
container_version: Optional[str] = None,

0 commit comments

Comments
 (0)