chore(risingwave): allow reuse of postgres get_schema

ibis-project · Aug 4, 2024 · b6b0e1d · b6b0e1d
1 parent 03f5dfe
commit b6b0e1d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/ibis/backends/postgres/__init__.py b/ibis/backends/postgres/__init__.py
@@ -304,7 +304,7 @@ def _post_connect(self) -> None:
             cur.execute("SET TIMEZONE = UTC")
 
     @property
-    def _session_temp_db(self):
+    def _session_temp_db(self) -> str | None:
         # Postgres doesn't assign the temporary table database until the first
         # temp table is created in a given session.
         # Before that temp table is created, this will return `None`

diff --git a/ibis/backends/risingwave/__init__.py b/ibis/backends/risingwave/__init__.py
@@ -586,3 +586,9 @@ def drop_sink(
         )
         with self._safe_raw_sql(src):
             pass
+
+    @property
+    def _session_temp_db(self) -> str | None:
+        # Return `None`, because RisingWave does not implement temp tables like
+        # Postgres
+        return None