Skip to content

Commit

Permalink
Add typing for bools.
Browse files Browse the repository at this point in the history
  • Loading branch information
vyasr committed Mar 25, 2021
1 parent 28ac5b0 commit afe29d7
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions python/cudf/cudf/_lib/labeling.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand All @@ -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 = \
Expand Down

0 comments on commit afe29d7

Please sign in to comment.