fix: Buggy SQL for postgres source (feast-dev#3424)

xaniasd · web-flow · commit 1ea100ef472a · 2023-01-14T23:58:28.000-08:00
* Fix get_table_column_names_and_types for postgres source

Signed-off-by: Dimitris Stafylarakis &lt;xanias@gmail.com&gt;

* make linter happy

Signed-off-by: Dimitris Stafylarakis &lt;xanias@gmail.com&gt;

Signed-off-by: Dimitris Stafylarakis &lt;xanias@gmail.com&gt;
diff --git a/sdk/python/feast/infra/offline_stores/contrib/postgres_offline_store/postgres_source.py b/sdk/python/feast/infra/offline_stores/contrib/postgres_offline_store/postgres_source.py
@@ -111,9 +111,7 @@ def get_table_column_names_and_types(
         self, config: RepoConfig
     ) -> Iterable[Tuple[str, str]]:
         with _get_conn(config.offline_store) as conn, conn.cursor() as cur:
-            cur.execute(
-                f"SELECT * FROM ({self.get_table_query_string()}) AS sub LIMIT 0"
-            )
+            cur.execute(f"SELECT * FROM {self.get_table_query_string()} AS sub LIMIT 0")
             return (
                 (c.name, pg_type_code_to_pg_type(c.type_code)) for c in cur.description
             )
