fix getstate in _InnerPredictor (#366)

d5bbaa26 · wxchan · Qiwei Ye · 0dbc421a · d5bbaa26
Commit d5bbaa26 authored Mar 27, 2017 by wxchan Committed by Qiwei Ye Mar 27, 2017
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

python-package/lightgbm/basic.py python-package/lightgbm/basic.py +5 -0

No files found.
--- a/python-package/lightgbm/basic.py
+++ b/python-package/lightgbm/basic.py
@@ -343,6 +343,11 @@ class _InnerPredictor(object):
        if self.__is_manage_handle:
            _safe_call(_LIB.LGBM_BoosterFree(self.handle))

+    def __getstate__(self):
+        this = self.__dict__.copy()
+        this.pop('handle', None)
+        return this
+
    def predict(self, data, num_iteration=-1,
                raw_score=False, pred_leaf=False, data_has_header=False,
                is_reshape=True):