Skip to content

Commit 8b7b81b

Browse files
author
Payton Staub
committed
Update api docs
1 parent accac70 commit 8b7b81b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/sagemaker/workflow/pipeline.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -240,8 +240,7 @@ def definition(self) -> str:
240240
"""Converts a request structure to string representation for workflow service calls."""
241241
request_dict = self.to_request()
242242
request_dict["PipelineExperimentConfig"] = interpolate(
243-
request_dict["PipelineExperimentConfig"],
244-
{}
243+
request_dict["PipelineExperimentConfig"], {}
245244
)
246245
callback_output_to_step_map = _map_callback_outputs(self.steps)
247246
request_dict["Steps"] = interpolate(
@@ -274,6 +273,7 @@ def interpolate(
274273
275274
Args:
276275
request_obj (RequestType): The request dict.
276+
callback_output_to_step_map (Dict[str, str]): A dict of output name -> step name.
277277
278278
Returns:
279279
RequestType: The request dict with Parameter values replaced by their expression.
@@ -287,6 +287,7 @@ def _interpolate(obj: Union[RequestType, Any], callback_output_to_step_map: Dict
287287
288288
Args:
289289
obj (Union[RequestType, Any]): The request dict.
290+
callback_output_to_step_map (Dict[str, str]): A dict of output name -> step name.
290291
"""
291292
if isinstance(obj, (Expression, Parameter, Properties)):
292293
return obj.expr

0 commit comments

Comments
 (0)