From a41b0308419c335f84eafa7ce1750a9348851b20 Mon Sep 17 00:00:00 2001 From: jeisenma Date: Fri, 15 Nov 2024 10:07:21 -0700 Subject: [PATCH] fixing return statement --- elm/wizard.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/elm/wizard.py b/elm/wizard.py index b8ecae5..942a256 100644 --- 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):