From e3dede9e00a7c9c04a8bd7a3a64f71ae85ee683a Mon Sep 17 00:00:00 2001 From: Oliver Holworthy <1216955+oliverholworthy@users.noreply.github.com> Date: Wed, 7 Jun 2023 12:02:27 +0100 Subject: [PATCH] Convert index to array so that assignment works later on --- nvtabular/ops/categorify.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nvtabular/ops/categorify.py b/nvtabular/ops/categorify.py index 7fbfa1de77..1cfdbf34a6 100644 --- 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: