diff --git a/python/cudf/cudf/core/column/column.py b/python/cudf/cudf/core/column/column.py index fbd5c86852d..223b196b2e3 100644 --- a/python/cudf/cudf/core/column/column.py +++ b/python/cudf/cudf/core/column/column.py @@ -1544,7 +1544,7 @@ def isnull(self) -> ColumnBase: with acquire_spill_lock(): result = type(self).from_pylibcudf( - plc.unary.is_null(self.to_pylibcudf(mode="read")) + plc.unary.is_null(self.plc_column) ) if self.dtype.kind == "f": diff --git a/python/cudf/cudf/io/csv.py b/python/cudf/cudf/io/csv.py index 2c3362138ea..5b7a7d1b7a2 100644 --- a/python/cudf/cudf/io/csv.py +++ b/python/cudf/cudf/io/csv.py @@ -471,7 +471,7 @@ def _plc_write_csv( if index else table._columns ) - columns = [col.to_pylibcudf(mode="read") for col in iter_columns] + columns = [col.plc_column for col in iter_columns] col_names = [] if header: table_names = ( diff --git a/python/cudf/cudf/io/json.py b/python/cudf/cudf/io/json.py index 1d9f2002430..68e91144b5c 100644 --- a/python/cudf/cudf/io/json.py +++ b/python/cudf/cudf/io/json.py @@ -341,9 +341,7 @@ def _plc_write_json( try: # TODO: TableWithMetadata expects list[ColumnNameSpec] but receives list[tuple[Hashable, Any]] tbl_w_meta = plc.io.TableWithMetadata( - plc.Table( - [col.to_pylibcudf(mode="read") for col in table._columns] - ), + plc.Table([col.plc_column for col in table._columns]), colnames, # type: ignore[arg-type] ) options = ( diff --git a/python/cudf/cudf/io/orc.py b/python/cudf/cudf/io/orc.py index e3c333d9652..6b812547427 100644 --- a/python/cudf/cudf/io/orc.py +++ b/python/cudf/cudf/io/orc.py @@ -454,9 +454,7 @@ def _plc_write_orc( if table.index is None else itertools.chain(table.index._columns, table._columns) ) - plc_table = plc.Table( - [col.to_pylibcudf(mode="read") for col in columns] - ) + plc_table = plc.Table([col.plc_column for col in columns]) tbl_meta = plc.io.types.TableInputMetadata(plc_table) for level, idx_name in enumerate(table._index.names): tbl_meta.column_metadata[level].set_name( @@ -464,9 +462,7 @@ def _plc_write_orc( ) num_index_cols_meta = table.index.nlevels else: - plc_table = plc.Table( - [col.to_pylibcudf(mode="read") for col in table._columns] - ) + plc_table = plc.Table([col.plc_column for col in table._columns]) tbl_meta = plc.io.types.TableInputMetadata(plc_table) num_index_cols_meta = 0 @@ -565,9 +561,7 @@ def write_table(self, table): else: cols_to_write = table._columns - self.writer.write( - plc.Table([col.to_pylibcudf(mode="read") for col in cols_to_write]) - ) + self.writer.write(plc.Table([col.plc_column for col in cols_to_write])) def close(self): if not self.initialized: @@ -581,15 +575,13 @@ def _initialize_chunked_state(self, table): """ num_index_cols_meta = 0 - plc_table = plc.Table( - [col.to_pylibcudf(mode="read") for col in table._columns] - ) + plc_table = plc.Table([col.plc_column for col in table._columns]) self.tbl_meta = plc.io.types.TableInputMetadata(plc_table) if self.index is not False: if isinstance(table.index, MultiIndex): plc_table = plc.Table( [ - col.to_pylibcudf(mode="read") + col.plc_column for col in itertools.chain( table.index._columns, table._columns ) @@ -603,7 +595,7 @@ def _initialize_chunked_state(self, table): if table.index.name is not None: plc_table = plc.Table( [ - col.to_pylibcudf(mode="read") + col.plc_column for col in itertools.chain( table.index._columns, table._columns ) diff --git a/python/cudf/cudf/io/parquet.py b/python/cudf/cudf/io/parquet.py index c02b2bf1208..98e689e0e95 100644 --- a/python/cudf/cudf/io/parquet.py +++ b/python/cudf/cudf/io/parquet.py @@ -122,9 +122,7 @@ def _plc_write_parquet( index is None and not isinstance(table.index, RangeIndex) ): columns = itertools.chain(table.index._columns, table._columns) - plc_table = plc.Table( - [col.to_pylibcudf(mode="read") for col in columns] - ) + plc_table = plc.Table([col.plc_column for col in columns]) tbl_meta = plc.io.types.TableInputMetadata(plc_table) for level, idx_name in enumerate(table.index.names): idx_name_str = ioutils._index_level_name( @@ -137,9 +135,7 @@ def _plc_write_parquet( tbl_meta.column_metadata[level].set_name(idx_name_str) num_index_cols_meta = table.index.nlevels else: - plc_table = plc.Table( - [col.to_pylibcudf(mode="read") for col in table._columns] - ) + plc_table = plc.Table([col.plc_column for col in table._columns]) tbl_meta = plc.io.types.TableInputMetadata(plc_table) num_index_cols_meta = 0 @@ -1814,13 +1810,9 @@ def write_table(self, table, partitions_info=None) -> None: table.index.name is not None or isinstance(table.index, MultiIndex) ): columns = itertools.chain(table.index._columns, table._columns) - plc_table = plc.Table( - [col.to_pylibcudf(mode="read") for col in columns] - ) + plc_table = plc.Table([col.plc_column for col in columns]) else: - plc_table = plc.Table( - [col.to_pylibcudf(mode="read") for col in table._columns] - ) + plc_table = plc.Table([col.plc_column for col in table._columns]) self.writer.write(plc_table, partitions_info) def close(self, metadata_file_path=None) -> np.ndarray | None: @@ -1852,15 +1844,13 @@ def _initialize_chunked_state( # Set the table_metadata num_index_cols_meta = 0 - plc_table = plc.Table( - [col.to_pylibcudf(mode="read") for col in table._columns] - ) + plc_table = plc.Table([col.plc_column for col in table._columns]) self.tbl_meta = plc.io.types.TableInputMetadata(plc_table) if self.index is not False: if isinstance(table.index, MultiIndex): plc_table = plc.Table( [ - col.to_pylibcudf(mode="read") + col.plc_column for col in itertools.chain( table.index._columns, table._columns ) @@ -1874,7 +1864,7 @@ def _initialize_chunked_state( if table.index.name is not None: plc_table = plc.Table( [ - col.to_pylibcudf(mode="read") + col.plc_column for col in itertools.chain( table.index._columns, table._columns )