[incubator-tvm] 01/01: filter out error features (#5952)

[tq...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git

commit 1097ba1a25a21af84b2e837935fc93b735a652ec
Author: Meteorix <lx...@gmail.com>
AuthorDate: Mon Oct 12 02:36:42 2020 +0800

    filter out error features (#5952)
---
 python/tvm/autotvm/tuner/xgboost_cost_model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/tvm/autotvm/tuner/xgboost_cost_model.py b/python/tvm/autotvm/tuner/xgboost_cost_model.py
index 7b9df1c..9fd19ec 100644
--- a/python/tvm/autotvm/tuner/xgboost_cost_model.py
+++ b/python/tvm/autotvm/tuner/xgboost_cost_model.py
@@ -245,6 +245,8 @@ class XGBoostCostModel(CostModel):
         else:
             raise RuntimeError("Invalid feature type: " + self.fea_type)
         res = pool.map(feature_extract_func, data)
+        # filter out None output, which is caused by extract exception
+        res = [r for r in res if r is not None]
 
         # filter out feature with different shapes
         fea_len = len(self._get_feature([0])[0])