vincbeck commented on code in PR #33219:
URL: https://github.com/apache/airflow/pull/33219#discussion_r1294989565
##########
airflow/providers/amazon/aws/operators/sagemaker.py:
##########
@@ -1523,3 +1524,246 @@ def execute(self, context: Context) -> str:
arn = ans["ExperimentArn"]
self.log.info("Experiment %s created successfully with ARN %s.",
self.name, arn)
return arn
+
+
+class SageMakerCreateNotebookOperator(BaseOperator):
+ """
+ Create a SageMaker notebook.
+
+ More information regarding parameters of this operator can be found here
+
https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker/client/create_notebook_instance.html.
+
+ .. seealso:
+ For more information on how to use this operator, take a look at the
guide:
+ :ref:`howto/operator:SageMakerCreateNotebookOperator`
+
+ :param instance_name: The name of the notebook instance.
+ :param instance_type: The type of instance to create.
+ :param role_arn: The Amazon Resource Name (ARN) of the IAM role that
SageMaker can assume to access
+ :param volume_size_in_gb: Size in GB of the EBS root device volume of the
notebook instance.
+ :param volume_kms_key_id: The KMS key ID for the EBS root device volume.
+ :param lifecycle_config_name: The name of the lifecycle configuration to
associate with the notebook
+ :param direct_internet_access: Whether to enable direct internet access
for the notebook instance.
+ :param root_access: Whether to give the notebook instance root access to
the Amazon S3 bucket.
+ :param wait_for_completion: Whether or not to wait for the notebook to be
InService before returning
+ :param create_instance_kwargs: Additional configuration options for the
create call.
+ :param aws_conn_id: The AWS connection ID to use.
+
+ This operator returns The ARN of the created notebook.
Review Comment:
```suggestion
:return: The ARN of the created notebook.
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: commits-unsubscr...@airflow.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
