@@ -37,14 +37,16 @@ def least_duration(
3737 items_with_durations = _get_items_with_durations (items , durations )
3838
3939 # add index of item in list
40- items_with_durations = [(* tup , i ) for i , tup in enumerate (items_with_durations )]
40+ items_with_durations_indexed = [
41+ (* tup , i ) for i , tup in enumerate (items_with_durations )
42+ ]
4143
4244 # sort in ascending order
4345 sorted_items_with_durations = sorted (
44- items_with_durations , key = lambda tup : tup [1 ], reverse = True
46+ items_with_durations_indexed , key = lambda tup : tup [1 ], reverse = True
4547 )
4648
47- selected : "List[List[nodes.Item]]" = [[] for i in range (splits )]
49+ selected : "List[List[Tuple[ nodes.Item, int] ]]" = [[] for i in range (splits )]
4850 deselected : "List[List[nodes.Item]]" = [[] for i in range (splits )]
4951 duration : "List[float]" = [0 for i in range (splits )]
5052
@@ -116,7 +118,9 @@ def duration_based_chunks(
116118 ]
117119
118120
119- def _get_items_with_durations (items , durations ):
121+ def _get_items_with_durations (
122+ items : "List[nodes.Item]" , durations : "Dict[str, float]"
123+ ) -> "List[Tuple[nodes.Item, float]]" :
120124 durations = _remove_irrelevant_durations (items , durations )
121125 avg_duration_per_test = _get_avg_duration_per_test (durations )
122126 items_with_durations = [
0 commit comments