Skip to content

Commit 657c7e0

Browse files
stacichoshenlongtang
authored andcommitted
fix: Update pipeline.py and selective_execution_config.py with small fixes (aws#1099)
1 parent b1d588c commit 657c7e0

File tree

2 files changed

+10
-10
lines changed

2 files changed

+10
-10
lines changed

src/sagemaker/workflow/pipeline.py

+9-9
Original file line numberDiff line numberDiff line change
@@ -428,8 +428,8 @@ def list_executions(
428428
sort_by (str): The field by which to sort results(CreationTime/PipelineExecutionArn).
429429
sort_order (str): The sort order for results (Ascending/Descending).
430430
max_results (int): The maximum number of pipeline executions to return in the response.
431-
next_token (str): If the result of the previous ListPipelineExecutions request was
432-
truncated, the response includes a NextToken. To retrieve the next set of pipeline
431+
next_token (str): If the result of the previous `ListPipelineExecutions` request was
432+
truncated, the response includes a `NextToken`. To retrieve the next set of pipeline
433433
executions, use the token in the next request.
434434
435435
Returns:
@@ -474,9 +474,9 @@ def build_parameters_from_execution(
474474
"""Gets the parameters from an execution, update with optional parameter value overrides.
475475
476476
Args:
477-
pipeline_execution_arn (str): The arn of the pipeline execution.
477+
pipeline_execution_arn (str): The arn of the reference pipeline execution.
478478
parameter_value_overrides (Dict[str, Union[str, bool, int, float]]): Parameter dict
479-
to be updated in the parameters from the referenced execution.
479+
to be updated with the parameters from the referenced execution.
480480
481481
Returns:
482482
A parameter dict built from an execution and provided parameter value overrides.
@@ -730,14 +730,14 @@ def list_parameters(self, max_results: int = None, next_token: str = None):
730730
731731
Args:
732732
max_results (int): The maximum number of parameters to return in the response.
733-
next_token (str): If the result of the previous ListPipelineParametersForExecution
734-
request was truncated, the response includes a NextToken. To retrieve the next
733+
next_token (str): If the result of the previous `ListPipelineParametersForExecution`
734+
request was truncated, the response includes a `NextToken`. To retrieve the next
735735
set of parameters, use the token in the next request.
736736
737737
Returns:
738-
Information about the parameters of the pipeline execution.
739-
See boto3 client list_pipeline_executions
740-
https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.list_pipeline_parameters_for_execution
738+
Information about the parameters of the pipeline execution. This function is also
739+
a wrapper for `list_pipeline_parameters_for_execution
740+
<https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.list_pipeline_parameters_for_execution>`_.
741741
"""
742742
kwargs = dict(PipelineExecutionArn=self.arn)
743743
update_args(

src/sagemaker/workflow/selective_execution_config.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def __init__(
3838
selected_steps (List[str]): A list of pipeline steps to run. All step(s) in all
3939
path(s) between two selected steps should be included.
4040
reference_latest_execution (bool): Whether to reference the latest execution if
41-
source_pipeline_execution_arn is not provided.
41+
`source_pipeline_execution_arn` is not provided.
4242
"""
4343
self.source_pipeline_execution_arn = source_pipeline_execution_arn
4444
self.selected_steps = selected_steps

0 commit comments

Comments
 (0)