small fix

ibis-project · Mar 12, 2024 · 5373f05 · 5373f05
1 parent 469f0ae
commit 5373f05
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/ibis/backends/risingwave/__init__.py b/ibis/backends/risingwave/__init__.py
@@ -284,9 +284,7 @@ def create_table(
                 cur.execute(insert_stmt)
 
             if overwrite:
-                cur.execute(
-                    sge.Drop(kind="TABLE", this=this, exists=True).sql(self.dialect)
-                )
+                self.drop_table(name, database=database, schema=schema, force=True)
                 cur.execute(
                     f"ALTER TABLE {table.sql(self.dialect)} RENAME TO {this.sql(self.dialect)}"
                 )
@@ -408,10 +406,8 @@ def create_materialized_view(
         this = sg.table(name, catalog=database, quoted=self.compiler.quoted)
         with self._safe_raw_sql(create_stmt) as cur:
             if overwrite:
-                cur.execute(
-                    sge.Drop(kind="MATERIALIZED VIEW", this=this, exists=True).sql(
-                        self.dialect
-                    )
+                self.drop_materialized_view(
+                    name, database=database, schema=schema, force=True
                 )
                 cur.execute(
                     f"ALTER MATERIALIZED VIEW {table.sql(self.dialect)} RENAME TO {this.sql(self.dialect)}"

diff --git a/poetry.lock b/poetry.lock