Upgrade to Py3.12 and fix tests

PiperOrigin-RevId: 721981221

Author: xingyousong

.github/workflows/ci.yml

@@ -15,7 +15,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.11"]  # 3.x disabled b/c of 3.12 test failures w/ GRPC.
+        python-version: ["3.12"]  # 3.x disabled b/c of 3.12 test failures w/ GRPC.
         suffix: ["core", "benchmarks", "algorithms", "clients", "pyglove", "raytune"]
         include:
           - suffix: "clients"

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

@@ -17,7 +17,7 @@ jobs:
     - name: Set up Python
       uses: actions/setup-python@v4
       with:
-        python-version: '3.11'
+        python-version: '3.12'
     - name: Install dependencies
       # NOTE: grpcio-tools needs to be periodically updated to support later Python versions.
       run: |

.github/workflows/pypi-publish.yml

@@ -15,7 +15,7 @@ jobs:
     - name: Set up Python
       uses: actions/setup-python@v4
       with:
-        python-version: '3.11'
+        python-version: '3.12'
     - name: Install dependencies
       # NOTE: grpcio-tools needs to be periodically updated to support later Python versions.
       run: |

vizier/_src/service/performance_test.py

@@ -18,9 +18,9 @@
 
 import multiprocessing.pool
 import time
-from absl import logging
 
-from vizier._src.service import constants
+from absl import logging
+# from vizier._src.service import constants
 from vizier._src.service import vizier_client
 from vizier._src.service import vizier_server
 from vizier.benchmarks import experimenters
@@ -36,28 +36,18 @@ class PerformanceTest(parameterized.TestCase):
   @classmethod
   def setUpClass(cls):
     super().setUpClass()
-    cls.server = vizier_server.DefaultVizierServer(
-        database_url=constants.SQL_MEMORY_URL
-    )
+    cls.server = vizier_server.DefaultVizierServer(database_url=None)
     vizier_client.environment_variables.server_endpoint = cls.server.endpoint
 
-  @parameterized.parameters(
-      (1, 10, 2),
-      (2, 10, 2),
-      (10, 10, 2),
-      (50, 5, 2),
-      (100, 5, 2),
-  )
+  @parameterized.parameters((1, 10), (2, 10), (10, 10), (50, 5), (100, 5))
   def test_multiple_clients_basic(
-      self, num_simultaneous_clients, num_trials_per_client, dimension
+      self, num_simultaneous_clients, num_trials_per_client
   ):
     def fn(client_id: int):
-      experimenter = experimenters.BBOBExperimenterFactory(
-          'Sphere', dimension
-      )()
+      experimenter = experimenters.BBOBExperimenterFactory('Sphere', 2)()
       problem_statement = experimenter.problem_statement()
       study_config = pyvizier.StudyConfig.from_problem(problem_statement)
-      study_config.algorithm = pyvizier.Algorithm.NSGA2
+      study_config.algorithm = pyvizier.Algorithm.RANDOM_SEARCH
 
       client = vizier_client.create_or_load_study(
           owner_id='my_username',

vizier/_src/service/vizier_server.py

@@ -22,6 +22,7 @@
 from concurrent import futures
 import datetime
 import time
+from typing import Optional
 
 import attr
 import grpc
@@ -48,7 +49,9 @@ class DefaultVizierServer:
   """
 
   _host: str = attr.field(default='localhost')
-  _database_url: str = attr.field(default=constants.SQL_LOCAL_URL, kw_only=True)
+  _database_url: Optional[str] = attr.field(
+      default=constants.SQL_LOCAL_URL, kw_only=True
+  )
   _policy_factory: pythia.PolicyFactory = attr.field(
       factory=service_policy_factory_lib.DefaultPolicyFactory, kw_only=True
   )