fix: linters

dvilelaf · May 31, 2024 · 274b07f · 274b07f
1 parent 2aa82ce
commit 274b07f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scripts/test_prompts.py b/scripts/test_prompts.py
@@ -31,7 +31,10 @@
 
 
 def timer(func):
+    """Timer decorator"""
+
     def wrapper(*args, **kwargs):
+        """Wrapper"""
         start_time = time.time()
         result = func(*args, **kwargs)
         end_time = time.time()
@@ -53,6 +56,7 @@ def wrapper(*args, **kwargs):
 
 @timer
 def run_llm():
+    """Run LLM"""
     output = llm.create_chat_completion(
         messages=[
             {"role": "system", "content": SYSTEM_PROMPTS[-1]},  # type: ignore