fix(tree): fix to_float (#873)

Co-authored-by: 杭卫强 <hangweiqiang@bytedance.com>
bytedance · Aug 10, 2021 · 0c25ce6 · 0c25ce6
1 parent eed77f2
commit 0c25ce6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fedlearner/model/tree/trainer.py b/fedlearner/model/tree/trainer.py
@@ -218,7 +218,7 @@ def read_data(file_type, filename, require_example_ids, require_labels,
     features = []
     cat_features = []
     def to_float(x):
-        float(x if x != '' else 'nan')
+        return float(x if x != '' else 'nan')
     for line in reader:
         if file_type == 'tfrecord':
             line = parse_tfrecord(line)