From f1165d8c023b4cf86a9fa176c51565496792d348 Mon Sep 17 00:00:00 2001 From: Oliver Holworthy Date: Wed, 7 Jun 2023 13:55:19 +0100 Subject: [PATCH 1/2] Convert index to array so that assignment works later on (#1836) --- 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: From 66c6e3a1b240b6b3addf96af5021a1c4f57e9d5b Mon Sep 17 00:00:00 2001 From: Oliver Holworthy Date: Wed, 7 Jun 2023 14:23:28 +0100 Subject: [PATCH 2/2] Remove n_workers=2 from Distributed in client fixture (#1835) --- tests/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index 0ca0ad8748..3c3ae4373b 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -97,7 +97,7 @@ def assert_eq(a, b, *args, **kwargs): @pytest.fixture(scope="module") def client(): - distributed = Distributed(n_workers=2) + distributed = Distributed() cluster = distributed.cluster client = distributed.client yield client