From c5130e487a3746089d2183dca799159c3d4bf941 Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Wed, 28 Nov 2018 09:31:46 -0800 Subject: [PATCH] Fix #3894: Allow loading pickles without self.booster attributes (redux) (#3944) --- python-package/xgboost/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python-package/xgboost/core.py b/python-package/xgboost/core.py index 0e312b538..60ba97a40 100644 --- a/python-package/xgboost/core.py +++ b/python-package/xgboost/core.py @@ -1444,7 +1444,7 @@ class Booster(object): One of the importance types defined above. """ - if self.booster != 'gbtree': + if getattr(self, 'booster', None) is not None and self.booster != 'gbtree': raise ValueError('Feature importance is not defined for Booster type {}' .format(self.booster))