added ability to support schemas other than public to `generate_pos…

…tgres_schema`
defog-ai · Apr 25, 2024 · 3b0cdaa · 3b0cdaa
1 parent 84d282d
commit 3b0cdaa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/defog/generate_schema.py b/defog/generate_schema.py
@@ -91,7 +91,7 @@ def generate_postgres_schema(
                 f"Please feel free to open a github issue at https://github.com/defog-ai/defog-python if this a generic library issue, or email [email protected]."
             )
     else:
-        return schemas
+        return table_columns
 
 
 def generate_redshift_schema(