Update python/cudf/cudf/core/udf/utils.py

Co-authored-by: Michael Wang <[email protected]>
rapidsai · Jan 27, 2022 · 643d55e · 643d55e
1 parent 2415a98
commit 643d55e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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