Skip to content

Commit 2da12b1

Browse files
committed
Fix typehints
1 parent 81af74b commit 2da12b1

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

src/pytest_split/algorithms.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)