diff --git a/ibis/backends/polars/__init__.py b/ibis/backends/polars/__init__.py index a87ff7683c70..c389e5e3f4cd 100644 --- a/ibis/backends/polars/__init__.py +++ b/ibis/backends/polars/__init__.py @@ -148,10 +148,6 @@ def _add_table(self, name: str, obj: pl.LazyFrame | pl.DataFrame) -> None: self._tables[name] = obj self._context.register(name, obj) - def _remove_table(self, name: str) -> None: - del self._tables[name] - self._context.unregister(name) - def sql( self, query: str, schema: sch.Schema | None = None, dialect: str | None = None ) -> ir.Table: @@ -407,6 +403,7 @@ def create_view( def drop_table(self, name: str, *, force: bool = False) -> None: if name in self._tables: del self._tables[name] + self._context.unregister(name) elif not force: raise com.IbisError(f"Table {name!r} does not exist") @@ -559,7 +556,7 @@ def _load_into_cache(self, name, expr): self.create_table(name, self.compile(expr).cache()) def _clean_up_cached_table(self, name): - self._remove_table(name) + self.drop_table(name, force=True) @lazy_singledispatch