'PostgreSQL' object has no attribute '_conn'

datafold · Oct 3, 2023 · 8f8cdf8 · 8f8cdf8
1 parent 943d128
commit 8f8cdf8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/data_diff/databases/postgresql.py b/data_diff/databases/postgresql.py
@@ -147,10 +147,10 @@ def create_connection(self):
 
         pg = import_postgresql()
         try:
-            c = pg.connect(**self._args)
+            self._conn = pg.connect(**self._args)
             if SESSION_TIME_ZONE:
-                c.cursor().execute(f"SET TIME ZONE '{SESSION_TIME_ZONE}'")
-            return c
+                self._conn.cursor().execute(f"SET TIME ZONE '{SESSION_TIME_ZONE}'")
+            return self._conn
         except pg.OperationalError as e:
             raise ConnectError(*e.args) from e