clear sample_weight_fields for predict

alibaba · Dec 13, 2024 · 2701194 · 2701194
1 parent 760221f
commit 2701194
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tzrec/main.py b/tzrec/main.py
@@ -1010,6 +1010,7 @@ def predict(
 
     data_config: DataConfig = pipeline_config.data_config
     data_config.ClearField("label_fields")
+    data_config.ClearField("sample_weight_fields")
     data_config.drop_remainder = False
     # Build feature
     features = _create_features(list(pipeline_config.feature_configs), data_config)