Merge pull request vanna-ai#434 from xtuhcy/main

Fixed a bug where the ask function did not return fig
marlonchen · May 20, 2024 · 6eb7f6b · 6eb7f6b
2 parents 4e844f3 + 7b6d210
commit 6eb7f6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py
@@ -1701,7 +1701,7 @@ def ask(
                 return None
             else:
                 return sql, None, None
-        return sql, df, None
+        return sql, df, fig
 
     def train(
         self,