Pass keyword argument for axis in dataframe any method (#1833)

NVIDIA-Merlin · Jun 5, 2023 · 3347981 · 3347981
1 parent 2ca6df0
commit 3347981
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nvtabular/ops/categorify.py b/nvtabular/ops/categorify.py
@@ -1025,7 +1025,7 @@ def _top_level_groupby(df, options: FitOptions = None, spill=True):
         del df_gb
 
         # Extract null groups into gb_null
-        isnull = gb.isnull().any(1)
+        isnull = gb.isnull().any(axis=1)
         gb_null = gb[~isnull]
         gb = gb[isnull]
         if not len(gb_null):