Fix FitCrossValidationCallback

DataCanvasIO · Oct 25, 2023 · f1a2add · f1a2add
1 parent ea7631d
commit f1a2add
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/hypergbm/hyper_gbm.py b/hypergbm/hyper_gbm.py
@@ -76,6 +76,17 @@ def on_search_end(self, hyper_model):
     def on_search_error(self, hyper_model):
         self.on_search_end(hyper_model)
 
+    def __getstate__(self):
+        try:
+            state = super().__getstate__()
+        except AttributeError:
+            state = self.__dict__
+
+        state = state.copy()
+        state['pbar'] = None
+
+        return state
+
 
 class HyperGBMExplainer:
     def __init__(self, hypergbm_estimator, data=None):