Allow for linear kernel for ARD in Python GP.

PiperOrigin-RevId: 589272278
google · Dec 9, 2023 · ec4c51c · ec4c51c
1 parent 097adc7
commit ec4c51c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vizier/_src/algorithms/designers/gp_bandit.py b/vizier/_src/algorithms/designers/gp_bandit.py
@@ -578,7 +578,7 @@ def from_problem(
   ) -> 'VizierGPBandit':
     rng = jax.random.PRNGKey(seed or 0)
     if problem.is_single_objective:
-      return cls(problem, rng=rng)
+      return cls(problem, rng=rng, linear_coef=1.0)
     else:
       num_objectives = len(
           problem.metric_information.of_type(vz.MetricType.OBJECTIVE)
@@ -600,6 +600,7 @@ def _scalarized_ucb(
       )
       return cls(
           problem,
+          linear_coef=1.0,
           scoring_function_factory=scoring_fn_factory,
           scoring_function_is_parallel=True,
           use_trust_region=False,