Skip to content

Commit 9dbd694

Browse files
authored
documentation: Update PipelineModel.register documentation (#4206)
1 parent 1f9def5 commit 9dbd694

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/sagemaker/pipeline.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -408,7 +408,8 @@ def register(
408408
validation. Values can be "All" or "None" (default: None).
409409
410410
Returns:
411-
A `sagemaker.model.ModelPackage` instance.
411+
If ``sagemaker_session`` is a ``PipelineSession`` instance, returns pipeline step
412+
arguments. Otherwise, returns ``None``
412413
"""
413414
for model in self.models:
414415
if model.model_data is None:

0 commit comments

Comments
 (0)