42 lines
1.5 KiB
YAML
42 lines
1.5 KiB
YAML
name: 'SageMaker - Private Workforce'
|
|
description: |
|
|
Private workforce in SageMaker
|
|
inputs:
|
|
- name: region
|
|
description: 'The region where the cluster launches.'
|
|
- name: team_name
|
|
description: 'The name of your work team.'
|
|
- name: description
|
|
description: 'A description of the work team.'
|
|
- name: user_pool
|
|
description: 'An identifier for a user pool. The user pool must be in the same region as the service that you are calling.'
|
|
- name: user_groups
|
|
description: 'An identifier for a user group.'
|
|
- name: client_id
|
|
description: 'An identifier for an application client. You must create the app client ID using Amazon Cognito.'
|
|
- name: sns_topic
|
|
description: 'The ARN for the SNS topic to which notifications should be published.'
|
|
default: ''
|
|
- name: tags
|
|
description: 'Key-value pairs to categorize AWS resources.'
|
|
default: '{}'
|
|
outputs:
|
|
- {name: workteam_arn, description: 'The ARN of the workteam.'}
|
|
implementation:
|
|
container:
|
|
image: carowang/kubeflow-pipeline-aws-sm:20190809-02
|
|
command: ['python']
|
|
args: [
|
|
workteam.py,
|
|
--region, {inputValue: region},
|
|
--team_name, {inputValue: team_name},
|
|
--description, {inputValue: description},
|
|
--user_pool, {inputValue: user_pool},
|
|
--user_groups, {inputValue: user_groups},
|
|
--client_id, {inputValue: client_id},
|
|
--sns_topic, {inputValue: sns_topic},
|
|
--tags, {inputValue: tags},
|
|
]
|
|
fileOutputs:
|
|
workteam_arn: /tmp/workteam_arn.txt
|