fixing lint

NREL · Nov 15, 2024 · b0429a4 · b0429a4
1 parent 785ec4d
commit b0429a4
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/elm/wizard.py b/elm/wizard.py
@@ -102,12 +102,10 @@ def engineer_query(self, query,
             query = '\n\n'.join(query)
 
         token_budget = token_budget or self.token_budget
-        if timeit:
-            start_time = perf_counter()
+        start_time = perf_counter()
         strings, _, idx = self.query_vector_db(query)
-        if timeit:
-            end_time = perf_counter()
-            vector_query_time = end_time - start_time
+        end_time = perf_counter()
+        vector_query_time = end_time - start_time
         message = copy.deepcopy(self.MODEL_INSTRUCTION)
         question = f"\n\nQuestion: {query}"
         used_index = []
@@ -253,7 +251,7 @@ def chat(self, query,
             "vectordb_query_time": vector_query_time
         }
         if timeit and debug:
-            return response_message, query, references, performance 
+            return response_message, query, references, performance
         if debug:
             return response_message, query, references
         else: