From 6582b65a86d83d71fd40ca07925302b824256df2 Mon Sep 17 00:00:00 2001 From: Sara Rabhi Date: Fri, 18 Nov 2022 20:49:27 -0500 Subject: [PATCH] fix failing ci error (#541) --- transformers4rec/torch/utils/data_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/transformers4rec/torch/utils/data_utils.py b/transformers4rec/torch/utils/data_utils.py index 17790e9f42..21907810f5 100644 --- 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(