From 2701194f519f875d1b0eb1025b5dc8598635930f Mon Sep 17 00:00:00 2001
From: gecheng <eric.gc@alibaba-inc.com>
Date: Fri, 13 Dec 2024 11:45:23 +0800
Subject: [PATCH] clear sample_weight_fields for predict

---
 tzrec/main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tzrec/main.py b/tzrec/main.py
index b176d4a..f722442 100644
--- 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)