Skip to content

Commit e2624af

Browse files
feature: Enable default role for Spark processors (#3801)
1 parent cb019f6 commit e2624af

File tree

1 file changed

+21
-21
lines changed

1 file changed

+21
-21
lines changed

src/sagemaker/spark/processing.py

Lines changed: 21 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -122,11 +122,11 @@ def __init__(
122122
framework_version (str): The version of SageMaker PySpark.
123123
py_version (str): The version of python.
124124
container_version (str): The version of spark container.
125-
role (str): An AWS IAM role name or ARN. The Amazon SageMaker training jobs
126-
and APIs that create Amazon SageMaker endpoints use this role
127-
to access training data and model artifacts. After the endpoint
128-
is created, the inference code might use the IAM role, if it
129-
needs to access an AWS resource.
125+
role (str): An AWS IAM role name or ARN. Amazon SageMaker Processing
126+
uses this role to access AWS resources, such as
127+
data stored in Amazon S3 (default: None).
128+
If not specified, the value from the defaults configuration file
129+
will be used.
130130
instance_type (str): Type of EC2 instance to use for
131131
processing, for example, 'ml.c4.xlarge'.
132132
instance_count (int): The number of instances to run
@@ -741,9 +741,9 @@ class PySparkProcessor(_SparkProcessorBase):
741741

742742
def __init__(
743743
self,
744-
role: str,
745-
instance_type: Union[str, PipelineVariable],
746-
instance_count: Union[int, PipelineVariable],
744+
role: str = None,
745+
instance_type: Union[str, PipelineVariable] = None,
746+
instance_count: Union[int, PipelineVariable] = None,
747747
framework_version: Optional[str] = None,
748748
py_version: Optional[str] = None,
749749
container_version: Optional[str] = None,
@@ -769,11 +769,11 @@ def __init__(
769769
framework_version (str): The version of SageMaker PySpark.
770770
py_version (str): The version of python.
771771
container_version (str): The version of spark container.
772-
role (str): An AWS IAM role name or ARN. The Amazon SageMaker training jobs
773-
and APIs that create Amazon SageMaker endpoints use this role
774-
to access training data and model artifacts. After the endpoint
775-
is created, the inference code might use the IAM role, if it
776-
needs to access an AWS resource.
772+
role (str): An AWS IAM role name or ARN. Amazon SageMaker Processing
773+
uses this role to access AWS resources, such as
774+
data stored in Amazon S3 (default: None).
775+
If not specified, the value from the defaults configuration file
776+
will be used.
777777
instance_type (str or PipelineVariable): Type of EC2 instance to use for
778778
processing, for example, 'ml.c4.xlarge'.
779779
instance_count (int or PipelineVariable): The number of instances to run
@@ -1018,9 +1018,9 @@ class SparkJarProcessor(_SparkProcessorBase):
10181018

10191019
def __init__(
10201020
self,
1021-
role: str,
1022-
instance_type: Union[str, PipelineVariable],
1023-
instance_count: Union[int, PipelineVariable],
1021+
role: str = None,
1022+
instance_type: Union[str, PipelineVariable] = None,
1023+
instance_count: Union[int, PipelineVariable] = None,
10241024
framework_version: Optional[str] = None,
10251025
py_version: Optional[str] = None,
10261026
container_version: Optional[str] = None,
@@ -1046,11 +1046,11 @@ def __init__(
10461046
framework_version (str): The version of SageMaker PySpark.
10471047
py_version (str): The version of python.
10481048
container_version (str): The version of spark container.
1049-
role (str): An AWS IAM role name or ARN. The Amazon SageMaker training jobs
1050-
and APIs that create Amazon SageMaker endpoints use this role
1051-
to access training data and model artifacts. After the endpoint
1052-
is created, the inference code might use the IAM role, if it
1053-
needs to access an AWS resource.
1049+
role (str): An AWS IAM role name or ARN. Amazon SageMaker Processing
1050+
uses this role to access AWS resources, such as
1051+
data stored in Amazon S3 (default: None).
1052+
If not specified, the value from the defaults configuration file
1053+
will be used.
10541054
instance_type (str or PipelineVariable): Type of EC2 instance to use for
10551055
processing, for example, 'ml.c4.xlarge'.
10561056
instance_count (int or PipelineVariable): The number of instances to run

0 commit comments

Comments
 (0)