Compare commits

..

6 Commits

Author SHA1 Message Date
Margaret
885e8db94a
Update split_workload.py 2023-08-09 17:50:12 +03:00
pre-commit-ci[bot]
a1bdd6e30c [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2023-08-09 14:44:02 +00:00
Margaret
1b79142b68
Adding type hints 2023-08-09 17:43:28 +03:00
Margaret
84636cc08c
Removing unused variable 2023-08-09 17:40:45 +03:00
Margaret
6ad14ebe6e
Fixing code 2023-08-09 17:37:51 +03:00
Margaret
e82d110c44
Fixing errors 2023-08-09 17:18:32 +03:00

View File

@ -1,9 +1,6 @@
from typing import List, Tuple, Union
def split_list(
timings: list[int | float | str],
) -> tuple[list[int | float], list[int | float], int | float]:
) -> tuple[list[int | float], list[int | float], int | float] | None:
"""
This algorithm is a brute-force search over (nearly) all 2^n
@ -30,35 +27,35 @@ def split_list(
ValueError: Timings must be a list of numbers
"""
for i, current_element in enumerate(timings):
valid_timings = list[int | float]
for current_element in enumerate(timings):
if (
isinstance(current_element, str)
and current_element.replace(".", "", 1).isdigit()
):
is_current_elem_int = float(current_element).is_integer()
if not is_current_elem_int:
timings[i] = abs(float(current_element))
valid_timings.append(abs(float(current_element)))
else:
timings[i] = abs(int(float(current_element)))
valid_timings.append(abs(int(float(current_element))))
elif isinstance(current_element, (int, float)):
timings[i] = abs(current_element)
valid_timings.append(abs(current_element))
else:
raise ValueError("Timings must be a list of numbers")
if len(timings) == 0:
if len(valid_timings) == 0:
return ([], [], 0)
elif len(timings) == 1:
return ([timings[0]], [], timings[0])
elif len(valid_timings) == 1:
return ([valid_timings[0]], [], valid_timings[0])
result = None
n = len(timings)
n = len(valid_timings)
smallest_diff = float("inf")
all_nums_positive = [c >= 0 for c in timings]
for i in range(1, 2**n - 1):
indices = [j for j in range(n) if i & (1 << j) != 0]
distributed_timings_2 = [timings[j] for j in range(n) if j not in indices]
diff = abs(sum(distributed_timings_1) - sum(distributed_timings_2))
distributed_timings_1 = [valid_timings[j] for j in indices]
distributed_timings_2 = [valid_timings[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 = (