Fix bug where zeta init arg was missing (#842)

secondmind-labs · May 16, 2024 · adcd419 · adcd419
1 parent f7f28da
commit adcd419
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/trieste/acquisition/rule.py b/trieste/acquisition/rule.py
@@ -1897,16 +1897,18 @@ def __init__(
         global_search_space: Box,
         beta: float = 0.7,
         kappa: float = 1e-4,
+        zeta: float = 0.5,
         min_eps: float = 1e-2,
         region_index: Optional[int] = None,
         input_active_dims: Optional[Union[slice, Sequence[int]]] = None,
     ):
         self._is_global = False
         super().__init__(
-            global_search_space,
-            beta,
-            kappa,
-            min_eps,
+            global_search_space=global_search_space,
+            beta=beta,
+            kappa=kappa,
+            zeta=zeta,
+            min_eps=min_eps,
             region_index=region_index,
             input_active_dims=input_active_dims,
         )