Skip to content

Commit 7fe9139

Browse files
committed
Add todo notes in pipeline class
1 parent 425f868 commit 7fe9139

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

  • sagemaker-mlops/src/sagemaker/mlops/workflow

sagemaker-mlops/src/sagemaker/mlops/workflow/pipeline.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,7 @@ def create(
171171
if self.sagemaker_session.local_mode:
172172
if parallelism_config:
173173
logger.warning("Pipeline parallelism config is not supported in the local mode.")
174+
# TODO: replace with sagemaker-core methods
174175
return self.sagemaker_session.sagemaker_client.create_pipeline(self, description)
175176
tags = format_tags(tags)
176177
tags = _append_project_tags(tags)
@@ -180,6 +181,7 @@ def create(
180181
kwargs,
181182
Tags=tags,
182183
)
184+
# TODO: replace with sagemaker-core methods
183185
return self.sagemaker_session.sagemaker_client.create_pipeline(**kwargs)
184186

185187
def _create_args(
@@ -510,6 +512,8 @@ def list_pipeline_versions(
510512
NextToken=next_token,
511513
MaxResults=max_results,
512514
)
515+
516+
# TODO: replace with sagemaker-core methods
513517
return self.sagemaker_session.sagemaker_client.list_pipeline_versions(**kwargs)
514518

515519
def _get_latest_execution_arn(self):

0 commit comments

Comments
 (0)