diff --git a/sagemaker-core/src/sagemaker/core/utils/__init__.py b/sagemaker-core/src/sagemaker/core/utils/__init__.py index 9947387537..3b3924c6d9 100644 --- a/sagemaker-core/src/sagemaker/core/utils/__init__.py +++ b/sagemaker-core/src/sagemaker/core/utils/__init__.py @@ -1,3 +1,5 @@ +# Test comment: Added per reviewer request as a test comment. + # 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..18f4692108 --- /dev/null +++ b/src/sagemaker/workflow/pipeline.py @@ -0,0 +1,29 @@ +# 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. + +# Test comment: This module handles pipeline upsert operations. Added per reviewer request. + +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)