forked from aws/sagemaker-python-sdk
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpipeline.py
More file actions
27 lines (22 loc) · 834 Bytes
/
pipeline.py
File metadata and controls
27 lines (22 loc) · 834 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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)