diff --git a/src/llm_benchmark/control/single.py b/src/llm_benchmark/control/single.py index 9a314e6..b99fd8d 100644 --- a/src/llm_benchmark/control/single.py +++ b/src/llm_benchmark/control/single.py @@ -12,10 +12,7 @@ def sum_range(n: int) -> int: Returns: int: Sum of range of numbers from 0 to n """ - arr = [] - for i in range(n): - arr.append(i) - return sum(arr) + return (n * (n - 1)) // 2 @staticmethod def max_list(v: List[int]) -> int: @@ -27,11 +24,7 @@ def max_list(v: List[int]) -> int: Returns: int: Maximum value in the vector """ - max_val = v[0] - for i in range(1, len(v)): - if v[i] > max_val: - max_val = v[i] - return max_val + return max(v) if v else 0 @staticmethod def sum_modulus(n: int, m: int) -> int: @@ -44,8 +37,7 @@ def sum_modulus(n: int, m: int) -> int: Returns: int: Sum of modulus of numbers from 0 to n """ - arr = [] - for i in range(n): - if i % m == 0: - arr.append(i) - return sum(arr) + sum_val = 0 + for i in range(0, n, m): + sum_val += i + return sum_val \ No newline at end of file