Merge branch 'rishabh/fix-install-schema' of github.com:defog-ai/defo…

…g-python into rishabh/fix-install-schema
defog-ai · Jan 12, 2024 · f10203b · f10203b
2 parents e1ac606 + 6f8a6df
commit f10203b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/defog/__init__.py b/defog/__init__.py
@@ -458,9 +458,9 @@ def generate_mysql_schema(
 
         if len(tables) == 0:
             # get all tables
-            table_schema = self.db_creds.get("database", "")
+            db_name = self.db_creds.get("database", "")
             cur.execute(
-                "SELECT table_name FROM information_schema.tables WHERE table_schema = '{table_schema}';"
+                f"SELECT table_name FROM information_schema.tables WHERE table_schema = '{db_name}';"
             )
             tables = [row[0] for row in cur.fetchall()]