diff --git a/src/llm_benchmark/control/single.py b/src/llm_benchmark/control/single.py index 9a314e6..5063176 100644 --- a/src/llm_benchmark/control/single.py +++ b/src/llm_benchmark/control/single.py @@ -5,17 +5,14 @@ class SingleForLoop: @staticmethod def sum_range(n: int) -> int: """Sum of range of numbers from 0 to n - + Args: n (int): Number to sum up to - + Returns: int: Sum of range of numbers from 0 to n """ - arr = [] - for i in range(n): - arr.append(i) - return sum(arr) + return sum(range(n)) @staticmethod def max_list(v: List[int]) -> int: @@ -48,4 +45,4 @@ def sum_modulus(n: int, m: int) -> int: for i in range(n): if i % m == 0: arr.append(i) - return sum(arr) + return sum(arr) \ No newline at end of file