def test_append_project_tags(tmpdir): config = tmpdir.join(".sagemaker-code-config") config.write( '{"sagemakerProjectId": "proj-1234", "sagemakerProjectName": "proj-name"}' ) working_dir = tmpdir.mkdir("sub") tags = _append_project_tags(None, working_dir) assert tags == [ { "Key": "sagemaker:project-id", "Value": "proj-1234" }, { "Key": "sagemaker:project-name", "Value": "proj-name" }, ] tags = _append_project_tags([{"Key": "a", "Value": "b"}], working_dir) assert tags == [ { "Key": "a", "Value": "b" }, { "Key": "sagemaker:project-id", "Value": "proj-1234" }, { "Key": "sagemaker:project-name", "Value": "proj-name" }, ]
def create( self, role_arn: str, description: str = None, tags: List[Dict[str, str]] = None, ) -> Dict[str, Any]: """Creates a Pipeline in the Pipelines service. Args: role_arn (str): The role arn that is assumed by the pipeline to create step artifacts. description (str): A description of the pipeline. tags (List[Dict[str, str]]): A list of {"Key": "string", "Value": "string"} dicts as tags. Returns: A response dict from the service. """ tags = _append_project_tags(tags) kwargs = self._create_args(role_arn, description) update_args( kwargs, Tags=tags, ) return self.sagemaker_session.sagemaker_client.create_pipeline( **kwargs)
def create( self, role_arn: str, description: str = None, tags: List[Dict[str, str]] = None, parallelism_config: ParallelismConfiguration = None, ) -> Dict[str, Any]: """Creates a Pipeline in the Pipelines service. Args: role_arn (str): The role arn that is assumed by the pipeline to create step artifacts. description (str): A description of the pipeline. tags (List[Dict[str, str]]): A list of {"Key": "string", "Value": "string"} dicts as tags. parallelism_config (Optional[ParallelismConfiguration]): Parallelism configuration that is applied to each of the executions of the pipeline. It takes precedence over the parallelism configuration of the parent pipeline. Returns: A response dict from the service. """ tags = _append_project_tags(tags) kwargs = self._create_args(role_arn, description, parallelism_config) update_args( kwargs, Tags=tags, ) return self.sagemaker_session.sagemaker_client.create_pipeline( **kwargs)