diff --git a/python/cudf/cudf/_lib/labeling.pyx b/python/cudf/cudf/_lib/labeling.pyx index d7ceb09dab5..1f721766f8f 100644 --- a/python/cudf/cudf/_lib/labeling.pyx +++ b/python/cudf/cudf/_lib/labeling.pyx @@ -4,6 +4,7 @@ import numpy as np from enum import IntEnum from libc.stdint cimport uint32_t +from libcpp.bool cimport bool as cbool from libcpp.memory cimport unique_ptr from libcpp.utility cimport move @@ -19,8 +20,8 @@ from cudf._lib.cpp.column.column_view cimport column_view # Note that the parameter input shadows a Python built-in in the local scope, # but I'm not too concerned about that since there's no use-case for actual # input in this context. -def label_bins(Column input, Column left_edges, left_inclusive, - Column right_edges, right_inclusive): +def label_bins(Column input, Column left_edges, cbool left_inclusive, + Column right_edges, cbool right_inclusive): cdef inclusive c_left_inclusive = \ inclusive.YES if left_inclusive else inclusive.NO cdef inclusive c_right_inclusive = \