Update console rendering

neuml · Jan 19, 2023 · 244ad1d · 244ad1d
1 parent 9edabdd
commit 244ad1d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/python/paperai/query.py b/src/python/paperai/query.py
@@ -243,7 +243,7 @@ def query(embeddings, db, query, topn, threshold):
 
         # Create console printer
         console = Console(soft_wrap=True)
-        with console.capture() as capture:
+        with console.capture() as output:
             # Print query
             console.print(f"[dark_orange]Query: {query}[/dark_orange]")
             console.print()
@@ -296,7 +296,7 @@ def query(embeddings, db, query, topn, threshold):
                 console.print()
 
         # Print console output
-        print(capture.get())
+        print(output.get())
 
     @staticmethod
     def run(query, topn=None, path=None, threshold=None):