From fc6dcba2b5564ad31738581c9701dd64e9721dce Mon Sep 17 00:00:00 2001
From: TurinTech Bot <bot@turintech.ai>
Date: Wed, 25 Sep 2024 12:57:34 +0000
Subject: [PATCH] Artemis Changes

---
 src/llm_benchmark/control/double.py | 17 ++---------------
 src/llm_benchmark/control/single.py | 11 ++++-------
 2 files changed, 6 insertions(+), 22 deletions(-)

diff --git a/src/llm_benchmark/control/double.py b/src/llm_benchmark/control/double.py
index 4be41d7..470651c 100644
--- a/src/llm_benchmark/control/double.py
+++ b/src/llm_benchmark/control/double.py
@@ -4,20 +4,7 @@
 class DoubleForLoop:
     @staticmethod
     def sum_square(n: int) -> int:
-        """Sum of squares of numbers from 0 to n (exclusive)
-
-        Args:
-            n (int): Number to sum up to
-
-        Returns:
-            int: Sum of squares of numbers from 0 to n
-        """
-        sum_ = 0
-        for i in range(n):
-            for j in range(n):
-                if i == j:
-                    sum_ += i * j
-        return sum_
+        return sum(i * i for i in range(n))
 
     @staticmethod
     def sum_triangle(n: int) -> int:
@@ -90,4 +77,4 @@ def sum_matrix(m: List[List[int]]) -> int:
         for i in range(len(m)):
             for j in range(len(m[i])):
                 sum_ += m[i][j]
-        return sum_
+        return sum_
\ No newline at end of file
diff --git a/src/llm_benchmark/control/single.py b/src/llm_benchmark/control/single.py
index 9a314e6..cf30b52 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 (n * (n + 1)) // 2
 
     @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