Skip to content

Commit fb34f93

Browse files
committed
fix: double Run create on load_run
fixes aws#3673
1 parent 4884d18 commit fb34f93

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

src/sagemaker/experiments/run.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -633,7 +633,8 @@ def _extract_run_name_from_tc_name(trial_component_name: str, experiment_name: s
633633
Returns:
634634
str: The name of the Run object supplied by a user.
635635
"""
636-
return trial_component_name.replace("{}{}".format(experiment_name, DELIMITER), "", 1)
636+
# TODO: we should revert the lower casting once backend fix reaches prod
637+
return trial_component_name.replace("{}{}".format(experiment_name.lower(), DELIMITER), "", 1)
637638

638639
@staticmethod
639640
def _append_run_tc_label_to_tags(tags: Optional[List[Dict[str, str]]] = None) -> list:
@@ -843,7 +844,6 @@ def list_runs(
843844
created_after: Optional[datetime.datetime] = None,
844845
sagemaker_session: Optional["Session"] = None,
845846
max_results: Optional[int] = None,
846-
next_token: Optional[str] = None,
847847
sort_by: SortByType = SortByType.CREATION_TIME,
848848
sort_order: SortOrderType = SortOrderType.DESCENDING,
849849
) -> list:
@@ -861,14 +861,15 @@ def list_runs(
861861
AWS services needed. If not specified, one is created using the
862862
default AWS configuration chain.
863863
max_results (int): Maximum number of Run objects to retrieve (default: None).
864-
next_token (str): Token for next page of results (default: None).
865864
sort_by (SortByType): The property to sort results by. One of NAME, CREATION_TIME
866865
(default: CREATION_TIME).
867866
sort_order (SortOrderType): One of ASCENDING, or DESCENDING (default: DESCENDING).
868867
869868
Returns:
870869
list: A list of ``Run`` objects.
871870
"""
871+
872+
# all trial components retrieved by default
872873
tc_summaries = _TrialComponent.list(
873874
experiment_name=experiment_name,
874875
created_before=created_before,
@@ -877,7 +878,6 @@ def list_runs(
877878
sort_order=sort_order.value,
878879
sagemaker_session=sagemaker_session,
879880
max_results=max_results,
880-
next_token=next_token,
881881
)
882882
run_list = []
883883
for tc_summary in tc_summaries:

tests/unit/sagemaker/experiments/test_run.py

-1
Original file line numberDiff line numberDiff line change
@@ -832,7 +832,6 @@ def test_list(mock_tc_search, mock_tc_list, mock_tc_load, run_obj, sagemaker_ses
832832
sort_order="Ascending",
833833
sagemaker_session=sagemaker_session,
834834
max_results=None,
835-
next_token=None,
836835
)
837836
assert len(run_list) == tc_list_len_half
838837
for i in range(tc_list_len_half):

0 commit comments

Comments
 (0)