Skip to content

Commit de02cde

Browse files
andre-marcos-perezclaytonparnell
authored andcommitted
fix: fix PySparkPySparkProcessor __init__ instance_type and instance_count params type
1 parent f2d5e41 commit de02cde

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
@@ -680,8 +680,8 @@ class PySparkProcessor(_SparkProcessorBase):
680680
def __init__(
681681
self,
682682
role: str,
683-
instance_type: Union[int, PipelineVariable],
684-
instance_count: Union[str, PipelineVariable],
683+
instance_type: Union[str, PipelineVariable],
684+
instance_count: Union[int, PipelineVariable],
685685
framework_version: Optional[str] = None,
686686
py_version: Optional[str] = None,
687687
container_version: Optional[str] = None,

0 commit comments

Comments
 (0)