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,9 +36,7 @@ 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(
@@ -61,7 +59,7 @@ def fn(client_id: int):
 
       client = vizier_client.create_or_load_study(
           owner_id='my_username',
-          study_id=self.id(),  # Use the testcase name.
+          study_id='my_study',
           study_config=study_config,
           client_id=str(client_id),
       )

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
   )