fixing return statement

NREL · Nov 15, 2024 · a41b030 · a41b030
1 parent b0429a4
commit a41b030
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/elm/wizard.py b/elm/wizard.py
@@ -130,8 +130,8 @@ def engineer_query(self, query,
         used_index = np.array(used_index)
         references = self.make_ref_list(used_index)
         if timeit:
-            return message, references, vector_query_time
-        return message, references
+            return message, references, used_index, vector_query_time
+        return message, references, used_index
 
     @abstractmethod
     def make_ref_list(self, idx):