@@ -633,7 +633,10 @@ def _extract_run_name_from_tc_name(trial_component_name: str, experiment_name: s
633
633
Returns:
634
634
str: The name of the Run object supplied by a user.
635
635
"""
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 (
638
+ "{}{}" .format (experiment_name .lower (), DELIMITER ), "" , 1
639
+ )
637
640
638
641
@staticmethod
639
642
def _append_run_tc_label_to_tags (tags : Optional [List [Dict [str , str ]]] = None ) -> list :
@@ -843,7 +846,6 @@ def list_runs(
843
846
created_after : Optional [datetime .datetime ] = None ,
844
847
sagemaker_session : Optional ["Session" ] = None ,
845
848
max_results : Optional [int ] = None ,
846
- next_token : Optional [str ] = None ,
847
849
sort_by : SortByType = SortByType .CREATION_TIME ,
848
850
sort_order : SortOrderType = SortOrderType .DESCENDING ,
849
851
) -> list :
@@ -861,14 +863,15 @@ def list_runs(
861
863
AWS services needed. If not specified, one is created using the
862
864
default AWS configuration chain.
863
865
max_results (int): Maximum number of Run objects to retrieve (default: None).
864
- next_token (str): Token for next page of results (default: None).
865
866
sort_by (SortByType): The property to sort results by. One of NAME, CREATION_TIME
866
867
(default: CREATION_TIME).
867
868
sort_order (SortOrderType): One of ASCENDING, or DESCENDING (default: DESCENDING).
868
869
869
870
Returns:
870
871
list: A list of ``Run`` objects.
871
872
"""
873
+
874
+ # all trial components retrieved by default
872
875
tc_summaries = _TrialComponent .list (
873
876
experiment_name = experiment_name ,
874
877
created_before = created_before ,
@@ -877,7 +880,6 @@ def list_runs(
877
880
sort_order = sort_order .value ,
878
881
sagemaker_session = sagemaker_session ,
879
882
max_results = max_results ,
880
- next_token = next_token ,
881
883
)
882
884
run_list = []
883
885
for tc_summary in tc_summaries :
0 commit comments