bugfix

vanna-ai · Aug 4, 2023 · e8e428f · e8e428f
1 parent 5443489
commit e8e428f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/vanna/__init__.py b/src/vanna/__init__.py
@@ -815,7 +815,7 @@ def train(question: str = None, sql: str = None, ddl: str = None, documentation:
 
     if documentation:
         print("Adding documentation....")
-        return add_documentation(sql)
+        return add_documentation(documentation)
 
     if sql:
         if question is None: