From 97115014fa9ff022c9366dd1325a07ccaa4ff4d3 Mon Sep 17 00:00:00 2001 From: Chansoo Lee Date: Thu, 28 Mar 2024 08:49:45 -0700 Subject: [PATCH] Add decorator `@seed_with_default` PiperOrigin-RevId: 619946318 --- optformer/vizier/data/filters.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/optformer/vizier/data/filters.py b/optformer/vizier/data/filters.py index 3b9d204..ad918f0 100644 --- a/optformer/vizier/data/filters.py +++ b/optformer/vizier/data/filters.py @@ -19,6 +19,7 @@ from optformer.common.data import filters from vizier import pyvizier as vz from vizier._src.pyglove import constants +from vizier._src.pyvizier.shared import parameter_iterators as pi from vizier.pyvizier import converters @@ -26,7 +27,7 @@ def _validate_parameters( parameters: dict[str, vz.ParameterValueTypes], space: vz.SearchSpace ) -> bool: """Call with parameters=trial.parameters.as_dict().""" - builder = vz.SequentialParameterBuilder(space, traverse_order='bfs') + builder = pi.SequentialParameterBuilder(space, traverse_order='bfs') try: for pc in builder: builder.choose_value(parameters.pop(pc.name))