diff --git a/sagemaker-core/src/sagemaker/core/utils/__init__.py b/sagemaker-core/src/sagemaker/core/utils/__init__.py index 9947387537..2dd9fb2eed 100644 --- a/sagemaker-core/src/sagemaker/core/utils/__init__.py +++ b/sagemaker-core/src/sagemaker/core/utils/__init__.py @@ -1,3 +1,4 @@ +# This is a test comment added as requested by the reviewer. # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"). You diff --git a/src/sagemaker/workflow/pipeline.py b/src/sagemaker/workflow/pipeline.py new file mode 100644 index 0000000000..c409b215f9 --- /dev/null +++ b/src/sagemaker/workflow/pipeline.py @@ -0,0 +1,27 @@ +# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 +# (the "License"). You may not use this file except in compliance +# with the License. A copy of the License is located at +# +# http://aws.amazon.com/apache2.0/ +# +# or in the "license" file accompanying this file. + +import logging + +logger = logging.getLogger(__name__) + + +class Pipeline: + def upsert(self, role_arn=None, **kwargs): + """Create or update the pipeline. + + Args: + role_arn (str): The ARN of the IAM role. + """ + if role_arn is not None and not isinstance(role_arn, str): + raise ValueError( + f"role_arn must be a string, got: {type(role_arn).__name__}" + ) + logger.info('Upserting pipeline with role_arn=%s', role_arn)