Skip to content

Commit 74ea72f

Browse files
xingyousongcopybara-github
authored andcommitted
1. Fix SQL datastore's commit logic
2. Small changes to performance_test.py (NSGA2 gives too many trials) 3. Allow Python 3.12 4. Move to 0.1.24 Vizier version PiperOrigin-RevId: 721981221
1 parent 615bb2a commit 74ea72f

File tree

8 files changed

+229
-195
lines changed

8 files changed

+229
-195
lines changed

.github/workflows/ci.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ jobs:
1515
strategy:
1616
fail-fast: false
1717
matrix:
18-
python-version: ["3.11"] # 3.x disabled b/c of 3.12 test failures w/ GRPC.
18+
python-version: ["3.12"] # 3.x disabled b/c of 3.13 test failures w/ JAX.
1919
suffix: ["core", "benchmarks", "algorithms", "clients", "pyglove", "raytune"]
2020
include:
2121
- suffix: "clients"

.github/workflows/pypi-publish-dev.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ jobs:
1717
- name: Set up Python
1818
uses: actions/setup-python@v4
1919
with:
20-
python-version: '3.11'
20+
python-version: '3.12'
2121
- name: Install dependencies
2222
# NOTE: grpcio-tools needs to be periodically updated to support later Python versions.
2323
run: |

.github/workflows/pypi-publish.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ jobs:
1515
- name: Set up Python
1616
uses: actions/setup-python@v4
1717
with:
18-
python-version: '3.11'
18+
python-version: '3.12'
1919
- name: Install dependencies
2020
# NOTE: grpcio-tools needs to be periodically updated to support later Python versions.
2121
run: |

vizier/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,4 +23,4 @@
2323

2424
sys.path.append(PROTO_ROOT)
2525

26-
__version__ = "0.1.23"
26+
__version__ = "0.1.24"

vizier/_src/service/performance_test.py

+5-13
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@
1818

1919
import multiprocessing.pool
2020
import time
21-
from absl import logging
2221

22+
from absl import logging
2323
from vizier._src.service import constants
2424
from vizier._src.service import vizier_client
2525
from vizier._src.service import vizier_server
@@ -41,23 +41,15 @@ def setUpClass(cls):
4141
)
4242
vizier_client.environment_variables.server_endpoint = cls.server.endpoint
4343

44-
@parameterized.parameters(
45-
(1, 10, 2),
46-
(2, 10, 2),
47-
(10, 10, 2),
48-
(50, 5, 2),
49-
(100, 5, 2),
50-
)
44+
@parameterized.parameters((1, 10), (2, 10), (10, 10), (50, 5), (100, 5))
5145
def test_multiple_clients_basic(
52-
self, num_simultaneous_clients, num_trials_per_client, dimension
46+
self, num_simultaneous_clients, num_trials_per_client
5347
):
5448
def fn(client_id: int):
55-
experimenter = experimenters.BBOBExperimenterFactory(
56-
'Sphere', dimension
57-
)()
49+
experimenter = experimenters.BBOBExperimenterFactory('Sphere', 2)()
5850
problem_statement = experimenter.problem_statement()
5951
study_config = pyvizier.StudyConfig.from_problem(problem_statement)
60-
study_config.algorithm = pyvizier.Algorithm.NSGA2
52+
study_config.algorithm = pyvizier.Algorithm.RANDOM_SEARCH
6153

6254
client = vizier_client.create_or_load_study(
6355
owner_id='my_username',

0 commit comments

Comments
 (0)