fix failing ci error (#541)

NVIDIA-Merlin · Nov 19, 2022 · 6582b65 · 6582b65
1 parent 7015a8e
commit 6582b65
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transformers4rec/torch/utils/data_utils.py b/transformers4rec/torch/utils/data_utils.py
@@ -314,7 +314,7 @@ def from_schema(
                 continuous_features or schema.select_by_tag(Tag.CONTINUOUS).column_names
             )
             targets = targets or schema.select_by_tag(Tag.TARGETS).column_names
-
+            schema = schema.select_by_name(categorical_features + continuous_features + targets)
             sparse_names = sparse_names or schema.select_by_tag(Tag.LIST).column_names
             sparse_max = sparse_max or {name: max_sequence_length for name in sparse_names}
             nvt_loader = cls(