From 187a43f9307811e1f5768a88d912f7f0d5c2f8e8 Mon Sep 17 00:00:00 2001 From: Matthew Murray Date: Mon, 25 Nov 2024 12:31:54 -0800 Subject: [PATCH] cpdef read_csv --- python/pylibcudf/pylibcudf/io/csv.pxd | 4 +++- python/pylibcudf/pylibcudf/io/csv.pyx | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/python/pylibcudf/pylibcudf/io/csv.pxd b/python/pylibcudf/pylibcudf/io/csv.pxd index 8515aa23b53..95f3ff4fe45 100644 --- a/python/pylibcudf/pylibcudf/io/csv.pxd +++ b/python/pylibcudf/pylibcudf/io/csv.pxd @@ -9,7 +9,7 @@ from pylibcudf.libcudf.io.csv cimport ( csv_reader_options, csv_reader_options_builder, ) -from pylibcudf.io.types cimport SinkInfo, SourceInfo +from pylibcudf.io.types cimport SinkInfo, SourceInfo, TableWithMetadata from pylibcudf.table cimport Table from pylibcudf.libcudf.io.types cimport ( @@ -61,6 +61,8 @@ cdef class CsvReaderOptionsBuilder: cpdef CsvReaderOptionsBuilder dayfirst(self, bool dayfirst) cpdef CsvReaderOptions build(self) +cpdef TableWithMetadata read_csv(CsvReaderOptions options) + cdef class CsvWriterOptions: cdef csv_writer_options c_obj cdef Table table diff --git a/python/pylibcudf/pylibcudf/io/csv.pyx b/python/pylibcudf/pylibcudf/io/csv.pyx index 0cafcfe8d1d..efc9bb813a1 100644 --- a/python/pylibcudf/pylibcudf/io/csv.pyx +++ b/python/pylibcudf/pylibcudf/io/csv.pyx @@ -628,7 +628,7 @@ cdef class CsvReaderOptionsBuilder: return csv_options -def read_csv( +cpdef TableWithMetadata read_csv( CsvReaderOptions options ): """