Merge pull request #37 from epinzur/fix_bug

fixed bug
epinzur · Jun 27, 2024 · 6c9c092 · 6c9c092
2 parents 09ed7c7 + f79d0ca
commit 6c9c092
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ragulate/pipelines/query_pipeline.py b/ragulate/pipelines/query_pipeline.py
@@ -95,10 +95,10 @@ def __init__(
                 queries = [queries[i] for i in sampled_indices]
 
             # Check for existing records and filter queries
-            existing_records = self._tru.get_records_and_feedback(
+            existing_records, _feedbacks = self._tru.get_records_and_feedback(
                 app_ids=[dataset.name]
             )
-            existing_queries = {record.query for record in existing_records}
+            existing_queries = existing_records["input"].dropna().tolist()
             queries = [query for query in queries if query not in existing_queries]
 
             self._queries[dataset.name] = queries