From 1daed600722915345183f246fae4437381900a64 Mon Sep 17 00:00:00 2001 From: vizier-team Date: Fri, 4 Aug 2023 15:42:34 -0700 Subject: [PATCH] [OSS Vizier] Tag `set_priors` in `gp_bandit.py` to allow experimental overrides. PiperOrigin-RevId: 553931922 --- vizier/_src/algorithms/designers/gp_bandit.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vizier/_src/algorithms/designers/gp_bandit.py b/vizier/_src/algorithms/designers/gp_bandit.py index ba6a0f005..970a065e8 100644 --- a/vizier/_src/algorithms/designers/gp_bandit.py +++ b/vizier/_src/algorithms/designers/gp_bandit.py @@ -209,6 +209,7 @@ def update( del all_active self._trials.extend(copy.deepcopy(completed.trials)) + @_experimental_override_allowed def set_priors(self, prior_studies: Sequence[vza.CompletedTrials]) -> None: """Updates the list of prior studies for transfer learning.