bayesian-optimization 3.2.0__tar.gz → 3.2.1__tar.gz

{bayesian_optimization-3.2.0 → bayesian_optimization-3.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: bayesian-optimization
-Version: 3.2.0
+Version: 3.2.1
 Summary: Bayesian Optimization package
 Author: Fernando Nogueira
 Author-email: Fernando Nogueira <fmfnogueira@gmail.com>

{bayesian_optimization-3.2.0 → bayesian_optimization-3.2.1}/bayes_opt/acquisition.py

@@ -99,13 +99,13 @@ class AcquisitionFunction(abc.ABC):
         )
         raise NotImplementedError(error_msg)
 
-    def set_acquisition_params(self, **params) -> None:
+    def set_acquisition_params(self, params: dict[str, Any]) -> None:
         """
         Set the parameters of the acquisition function.
 
         Parameters
         ----------
-        **params : dict
+        params : dict
             The parameters of the acquisition function.
         """
         error_msg = (
@@ -553,7 +553,7 @@ class UpperConfidenceBound(AcquisitionFunction):
         ):
             self.kappa = self.kappa * self.exploration_decay
 
-    def get_acquisition_params(self) -> dict:
+    def get_acquisition_params(self) -> dict[str, Any]:
         """Get the current acquisition function parameters.
 
         Returns
@@ -567,7 +567,7 @@ class UpperConfidenceBound(AcquisitionFunction):
             "exploration_decay_delay": self.exploration_decay_delay,
         }
 
-    def set_acquisition_params(self, params: dict) -> None:
+    def set_acquisition_params(self, params: dict[str, Any]) -> None:
         """Set the acquisition function parameters.
 
         Parameters
@@ -733,7 +733,7 @@ class ProbabilityOfImprovement(AcquisitionFunction):
         ):
             self.xi = self.xi * self.exploration_decay
 
-    def get_acquisition_params(self) -> dict:
+    def get_acquisition_params(self) -> dict[str, Any]:
         """Get the current acquisition function parameters.
 
         Returns
@@ -747,7 +747,7 @@ class ProbabilityOfImprovement(AcquisitionFunction):
             "exploration_decay_delay": self.exploration_decay_delay,
         }
 
-    def set_acquisition_params(self, params: dict) -> None:
+    def set_acquisition_params(self, params: dict[str, Any]) -> None:
         """Set the acquisition function parameters.
 
         Parameters
@@ -922,7 +922,7 @@ class ExpectedImprovement(AcquisitionFunction):
         ):
             self.xi = self.xi * self.exploration_decay
 
-    def get_acquisition_params(self) -> dict:
+    def get_acquisition_params(self) -> dict[str, Any]:
         """Get the current acquisition function parameters.
 
         Returns
@@ -936,7 +936,7 @@ class ExpectedImprovement(AcquisitionFunction):
             "exploration_decay_delay": self.exploration_decay_delay,
         }
 
-    def set_acquisition_params(self, params: dict) -> None:
+    def set_acquisition_params(self, params: dict[str, Any]) -> None:
         """Set the acquisition function parameters.
 
         Parameters
@@ -1147,7 +1147,7 @@ class ConstantLiar(AcquisitionFunction):
 
         return x_max
 
-    def get_acquisition_params(self) -> dict:
+    def get_acquisition_params(self) -> dict[str, Any]:
         """Get the current acquisition function parameters.
 
         Returns
@@ -1163,7 +1163,7 @@ class ConstantLiar(AcquisitionFunction):
             "rtol": self.rtol,
         }
 
-    def set_acquisition_params(self, params: dict) -> None:
+    def set_acquisition_params(self, params: dict[str, Any]) -> None:
         """Set the acquisition function parameters.
 
         Parameters
@@ -1318,7 +1318,7 @@ class GPHedge(AcquisitionFunction):
         idx = self._sample_idx_from_softmax_gains(random_state=random_state)
         return x_max[idx]
 
-    def get_acquisition_params(self) -> dict:
+    def get_acquisition_params(self) -> dict[str, Any]:
         """Get the current acquisition function parameters.
 
         Returns
@@ -1334,7 +1334,7 @@ class GPHedge(AcquisitionFunction):
             else None,
         }
 
-    def set_acquisition_params(self, params: dict) -> None:
+    def set_acquisition_params(self, params: dict[str, Any]) -> None:
         """Set the acquisition function parameters.
 
         Parameters

{bayesian_optimization-3.2.0 → bayesian_optimization-3.2.1}/bayes_opt/bayesian_optimization.py

@@ -328,7 +328,7 @@ class BayesianOptimization:
                 self._space.keys, self._space.res()[-1], self._space.params_config, self.max
             )
 
-    def random_sample(self, n: int = 1) -> dict[str, float | NDArray[Float]]:
+    def random_sample(self, n: int = 1) -> list[dict[str, float | NDArray[Float]]]:
         """Generate a random sample of parameters from the target space.
 
         Parameters
@@ -442,13 +442,13 @@ class BayesianOptimization:
         """
         random_state = None
         if self._random_state is not None:
-            state_tuple = self._random_state.get_state()
+            state_dict = self._random_state.get_state(legacy=False)
             random_state = {
-                "bit_generator": state_tuple[0],
-                "state": state_tuple[1].tolist(),
-                "pos": state_tuple[2],
-                "has_gauss": state_tuple[3],
-                "cached_gaussian": state_tuple[4],
+                "bit_generator": state_dict["bit_generator"],
+                "state": state_dict["state"]["key"].tolist(),
+                "pos": state_dict["state"]["pos"],
+                "has_gauss": state_dict["has_gauss"],
+                "cached_gaussian": state_dict["gauss"],
             }
 
         # Get constraint values if they exist

{bayesian_optimization-3.2.0 → bayesian_optimization-3.2.1}/bayes_opt/target_space.py

@@ -643,7 +643,7 @@ class TargetSpace:
         params = self.params[self.mask]
         target_max_idx = np.argmax(target)
 
-        res = {"target": target_max, "params": dict(zip(self.keys, params[target_max_idx]))}
+        res = {"target": target_max, "params": self.array_to_params(params[target_max_idx])}
 
         if self._constraint is not None:
             constraint_values = self.constraint_values[self.mask]
@@ -672,7 +672,7 @@ class TargetSpace:
 
             return [{"target": target, "params": param} for target, param in zip(self.target, params)]
 
-        params = [dict(zip(self.keys, p)) for p in self.params]
+        params = [self.array_to_params(p) for p in self.params]
 
         return [
             {"target": target, "constraint": constraint_value, "params": param, "allowed": allowed}

{bayesian_optimization-3.2.0 → bayesian_optimization-3.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bayesian-optimization"
-version = "3.2.0"
+version = "3.2.1"
 description = "Bayesian Optimization package"
 authors = [{ name = "Fernando Nogueira", email = "fmfnogueira@gmail.com" }]
 license = { file = "LICENSE" }