Update python/cudf/cudf/core/column/lists.py

Co-authored-by: Bradley Dice <bdice@bradleydice.com>
shwina · Apr 1, 2022 · 2cb1df3 · 2cb1df3
1 parent 01942e6
commit 2cb1df3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cudf/cudf/core/column/lists.py b/python/cudf/cudf/core/column/lists.py
@@ -381,7 +381,7 @@ def get(
         """
         out = extract_element(self._column, index)
 
-        if not (default is None or default is cudf.NA):
+        if default not in {None, cudf.NA}:
             # determine rows for which `index` is out-of-bounds
             lengths = count_elements(self._column)
             out_of_bounds_indexes = (-index > lengths) | (index >= lengths)