From a0ea8ac0957ddbb31084a52606668740d4e4426e Mon Sep 17 00:00:00 2001 From: TurinTech Bot Date: Fri, 20 Sep 2024 09:05:58 +0000 Subject: [PATCH] Artemis Changes --- src/llm_benchmark/algorithms/sort.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/src/llm_benchmark/algorithms/sort.py b/src/llm_benchmark/algorithms/sort.py index 5f36289..f3ded47 100644 --- a/src/llm_benchmark/algorithms/sort.py +++ b/src/llm_benchmark/algorithms/sort.py @@ -48,12 +48,6 @@ def max_n(v: List[int], n: int) -> List[int]: tmp = v.copy() ret = [-maxsize - 1] * n for i in range(n): - max_val = tmp[0] - max_idx = 0 - for j in range(1, len(tmp)): - if tmp[j] > max_val: - max_val = tmp[j] - max_idx = j - ret[i] = max_val - tmp.pop(max_idx) - return ret + max_idx = max(range(len(tmp)), key=tmp.__getitem__) + ret[i] = tmp[max_idx] + tmp.pop(max_idx) \ No newline at end of file