diff --git a/vizier/_src/benchmarks/experimenters/infeasible_experimenter.py b/vizier/_src/benchmarks/experimenters/infeasible_experimenter.py index e34b91ebd..0bd175dc3 100644 --- a/vizier/_src/benchmarks/experimenters/infeasible_experimenter.py +++ b/vizier/_src/benchmarks/experimenters/infeasible_experimenter.py @@ -60,7 +60,7 @@ def problem_statement(self) -> vz.ProblemStatement: @attrs.define -class ParameterRegionInfeasibleExperimenter(experimenter.Experimenter): +class ParamRegionInfeasibleExperimenter(experimenter.Experimenter): """Selects a parameter and splits its values into feasible/infeasible.""" exptr: experimenter.Experimenter = attrs.field() diff --git a/vizier/_src/benchmarks/experimenters/infeasible_experimenter_test.py b/vizier/_src/benchmarks/experimenters/infeasible_experimenter_test.py index b29f7ad03..b4f85529b 100644 --- a/vizier/_src/benchmarks/experimenters/infeasible_experimenter_test.py +++ b/vizier/_src/benchmarks/experimenters/infeasible_experimenter_test.py @@ -43,13 +43,13 @@ def test_consistency(self): ) -class ParameterRegionInfeasibleExperimenterTest(absltest.TestCase): +class ParamRegionInfeasibleExperimenterTest(absltest.TestCase): def test_e2e(self): exptr = numpy_experimenter.NumpyExperimenter( bbob.Sphere, bbob.DefaultBBOBProblemStatement(2) ) - exptr = infeasible_experimenter.ParameterRegionInfeasibleExperimenter( + exptr = infeasible_experimenter.ParamRegionInfeasibleExperimenter( exptr, parameter_name='x0', infeasible_interval=(0.0, 0.5) ) diff --git a/vizier/benchmarks/experimenters/__init__.py b/vizier/benchmarks/experimenters/__init__.py index 27975638f..5a0d82639 100644 --- a/vizier/benchmarks/experimenters/__init__.py +++ b/vizier/benchmarks/experimenters/__init__.py @@ -28,6 +28,8 @@ from vizier._src.benchmarks.experimenters.experimenter_factory import ExperimenterFactory from vizier._src.benchmarks.experimenters.experimenter_factory import SerializableExperimenterFactory from vizier._src.benchmarks.experimenters.experimenter_factory import SingleObjectiveExperimenterFactory +from vizier._src.benchmarks.experimenters.infeasible_experimenter import HashingInfeasibleExperimenter +from vizier._src.benchmarks.experimenters.infeasible_experimenter import ParamRegionInfeasibleExperimenter from vizier._src.benchmarks.experimenters.l1_categorical_experimenter import L1CategorialExperimenter from vizier._src.benchmarks.experimenters.multiobjective_experimenter import MultiObjectiveExperimenter from vizier._src.benchmarks.experimenters.noisy_experimenter import NoisyExperimenter