From 0d6a59d9aafbfe6d0e69e1c5a2f09367e12e0e11 Mon Sep 17 00:00:00 2001 From: TurinTech Bot Date: Tue, 8 Oct 2024 12:58:04 +0000 Subject: [PATCH] Artemis Changes --- src/llm_benchmark/control/single.py | 24 ++++++++++-------------- 1 file changed, 10 insertions(+), 14 deletions(-) diff --git a/src/llm_benchmark/control/single.py b/src/llm_benchmark/control/single.py index 9a314e6..b290084 100644 --- a/src/llm_benchmark/control/single.py +++ b/src/llm_benchmark/control/single.py @@ -35,17 +35,13 @@ def max_list(v: List[int]) -> int: @staticmethod def sum_modulus(n: int, m: int) -> int: - """Sum of modulus of numbers from 0 to n - - Args: - n (int): Number to sum up to - m (int): Modulus - - 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 of modulus of numbers from 0 to n + + Args: + n (int): Number to sum up to + m (int): Modulus + + Returns: + int: Sum of modulus of numbers from 0 to n + """ + return sum(i for i in range(n) if i % m == 0) \ No newline at end of file