Convert index to array so that assignment works later on (#1836)

NVIDIA-Merlin · Jun 7, 2023 · f1165d8 · f1165d8
1 parent 69fbd57
commit f1165d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nvtabular/ops/categorify.py b/nvtabular/ops/categorify.py
@@ -1693,7 +1693,7 @@ def _encode(
     expr = df[selection_l.names[0]].isna()
     for _name in selection_l.names[1:]:
         expr = expr & df[_name].isna()
-    nulls = df[expr].index
+    nulls = df[expr].index.values
 
     if use_collection or not search_sorted:
         if list_col: