cpdef read_csv

rapidsai · Nov 25, 2024 · 187a43f · 187a43f
1 parent 739e61f
commit 187a43f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -628,7 +628,7 @@ cdef class CsvReaderOptionsBuilder:
         return csv_options
 
 
-def read_csv(
+cpdef TableWithMetadata read_csv(
     CsvReaderOptions options
 ):
     """