diff --git a/fog_x/database/polars_connector.py b/fog_x/database/polars_connector.py index e53cd45..4bece21 100644 --- a/fog_x/database/polars_connector.py +++ b/fog_x/database/polars_connector.py @@ -146,9 +146,7 @@ def load_tables(self, table_names: List[str]): logger.debug(f"Table {table_name} does not exist in {path}.") def save_table(self, table_name: str): - self.tables[table_name].write_parquet( - f"{self.path}/{table_name}.parquet" - ) + pq.write_table(self.tables[table_name].to_arrow(), f"{self.path}/{table_name}.parquet") class LazyFrameConnector(PolarsConnector):