From 643d55e15d695dec385ceed83c686440ddbb1d33 Mon Sep 17 00:00:00 2001 From: brandon-b-miller <53796099+brandon-b-miller@users.noreply.github.com> Date: Thu, 27 Jan 2022 13:34:20 -0600 Subject: [PATCH] Update python/cudf/cudf/core/udf/utils.py Co-authored-by: Michael Wang --- python/cudf/cudf/core/udf/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/cudf/cudf/core/udf/utils.py b/python/cudf/cudf/core/udf/utils.py index b54b7d3bcf6..a98ee40274e 100644 --- a/python/cudf/cudf/core/udf/utils.py +++ b/python/cudf/cudf/core/udf/utils.py @@ -144,6 +144,7 @@ def _construct_signature(frame, return_type, args): @cuda.jit(device=True) def _mask_get(mask, pos): + """Return the validity of mask[pos] as a word.""" return (mask[pos // MASK_BITSIZE] >> (pos % MASK_BITSIZE)) & 1