Merge branch 'new_api' of github.com:NVIDIA/NVTabular into new_api

NVIDIA-Merlin · Dec 16, 2020 · 5467d86 · 5467d86
2 parents b21ffd8 + 6a84e7d
commit 5467d86
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/nvtabular/workflow.py b/nvtabular/workflow.py
@@ -24,8 +24,6 @@
 from nvtabular.ops import StatOperator
 from nvtabular.worker import clean_worker_cache
 
-# import yaml
-
 
 LOG = logging.getLogger("nvtabular")
 
@@ -158,7 +156,8 @@ def _transform_ddf(ddf, column_groups):
     columns = list(flatten(cg.flattened_columns for cg in column_groups))
 
     return ddf.map_partitions(
-        lambda gdf: _transform_partition(gdf, column_groups),
+        _transform_partition,
+        column_groups,
         meta=cudf.DataFrame({k: [] for k in columns}),
     )