v0.1.2 fix

Adjust __get_state to ignore timer as class attribute to make it work with dask
automl · Jul 11, 2024 · 58f9fb9 · 58f9fb9
1 parent bfc457a
commit 58f9fb9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/dehb/optimizers/dehb.py b/src/dehb/optimizers/dehb.py
@@ -269,6 +269,7 @@ def __getstate__(self):
         d = dict(self.__dict__)
         d["client"] = None  # hack to allow Dask client to be a class attribute
         d["logger"] = None  # hack to allow logger object to be a class attribute
+        d["_runtime_budget_timer"] = None # hack to allow timer object to be a class attribute
         return d
 
     def __del__(self):