Skip to content

Commit cd2a305

Browse files
ivan-khvostishkovnavinsoni
authored andcommitted
Fixing type hints for Spark processor that has instance type/count params in reverse order
1 parent ff0d9ae commit cd2a305

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/sagemaker/spark/processing.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -700,8 +700,8 @@ class PySparkProcessor(_SparkProcessorBase):
700700
def __init__(
701701
self,
702702
role: str,
703-
instance_type: Union[int, PipelineVariable],
704-
instance_count: Union[str, PipelineVariable],
703+
instance_type: Union[str, PipelineVariable],
704+
instance_count: Union[int, PipelineVariable],
705705
framework_version: Optional[str] = None,
706706
py_version: Optional[str] = None,
707707
container_version: Optional[str] = None,
@@ -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)