fix: add postgres schema name in create engine

datachecks · Aug 25, 2023 · 575f3ae · 575f3ae
1 parent a0e71c5
commit 575f3ae
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/datachecks/integrations/databases/postgres.py b/datachecks/integrations/databases/postgres.py
@@ -37,7 +37,12 @@ def connect(self) -> Any:
                 port=self.data_connection.get("port"),
                 database=self.data_connection.get("database"),
             )
-            engine = create_engine(url)
+            schema = self.data_connection.get("schema")
+            engine = create_engine(
+                url,
+                connect_args={"options": f"-csearch_path={schema}"},
+                isolation_level="AUTOCOMMIT",
+            )
             self.connection = engine.connect()
             return self.connection
         except Exception as e: