Update llama_index.py

lancedb · Oct 14, 2024 · c3b875f · c3b875f
1 parent bd510cc
commit c3b875f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ragged/dataset/llama_index.py b/ragged/dataset/llama_index.py
@@ -51,7 +51,7 @@ def get_contexts(self) -> List[TextNode]:
         return self.documents
 
     def get_queries(self) -> List[str]:
-        return self.dataset[self.query_column_name].tolist()
+        return self.to_pandas()[self.query_column_name].tolist()
 
     @property
     def context_column_name(self):