Updating cheduling/split_workload.py

implementing suggestion

Co-authored-by: Tianyi Zheng <tianyizheng02@gmail.com>

scheduling/split_workload.py

@@ -28,22 +28,20 @@ def split_list(timings: list) -> tuple:
 
     def split_workload(arr: list) -> tuple:
         if len(arr) == 0:
-            result = ([], [], 0)
-        elif len(arr) == 1:
-            result = ([arr[0]], [], 1)
+            return ([], [], 0)
+        if len(arr) == 1:
+            return ([arr[0]], [], 1)
         else:
             n = len(arr)
             smallest_diff = float("inf")
-            all_nums_positive = [c >= 0 for c in arr]
-            if False in all_nums_positive:
-                raise ValueError("numbers can only be positive")
+            if any(c < 0 for c in arr):
+                raise ValueError("Numbers can only be non-negative")
             for i in range(1, 2**n - 1):
                 indices = [j for j in range(n) if (i & (1 << j)) != 0]
                 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
+                smallest_diff = min(smallest_diff, diff)
                 result = (
                     distributed_timings_1,
                     distributed_timings_2,