From d053352d957992081733a3aafb60fae8d2f2551d Mon Sep 17 00:00:00 2001 From: Margaret <62753112+meg-1@users.noreply.github.com> Date: Wed, 26 Jul 2023 17:16:36 +0300 Subject: [PATCH] Updating cheduling/split_workload.py implementing suggestion Co-authored-by: Tianyi Zheng --- scheduling/split_workload.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/scheduling/split_workload.py b/scheduling/split_workload.py index e0a7a4853..ce25218cc 100644 --- a/scheduling/split_workload.py +++ b/scheduling/split_workload.py @@ -41,13 +41,12 @@ def split_list(timings: list) -> tuple: distributed_timings_1 = [arr[j] for j in indices] distributed_timings_2 = [arr[j] for j in range(n) if j not in indices] diff = abs(sum(distributed_timings_1) - sum(distributed_timings_2)) - if diff < smallest_diff: - smallest_diff = diff - result = ( - distributed_timings_1, - distributed_timings_2, - smallest_diff, - ) + smallest_diff = min(smallest_diff, diff) + result = ( + distributed_timings_1, + distributed_timings_2, + smallest_diff, + ) return result try: