From 78df035e44db151f7f6907323933e7c2797f6fad Mon Sep 17 00:00:00 2001
From: Ayush Agrawal <ayushagra@google.com>
Date: Thu, 23 Oct 2025 15:32:19 -0700
Subject: [PATCH] feat: Add Python 3.13 Kokoro run config

PiperOrigin-RevId: 823212081
---
 .kokoro/presubmit/unit_3-13.cfg        | 13 +++++++++++++
 CONTRIBUTING.rst                       |  4 +++-
 setup.py                               |  5 ++++-
 testing/constraints-3.13.txt           |  3 ++-
 tests/unit/vertexai/test_evaluation.py | 10 ++++++++++
 5 files changed, 32 insertions(+), 3 deletions(-)
 create mode 100644 .kokoro/presubmit/unit_3-13.cfg

diff --git a/.kokoro/presubmit/unit_3-13.cfg b/.kokoro/presubmit/unit_3-13.cfg
new file mode 100644
index 0000000000..f9403bf2d1
--- /dev/null
+++ b/.kokoro/presubmit/unit_3-13.cfg
@@ -0,0 +1,13 @@
+# Format: //devtools/kokoro/config/proto/build.proto
+
+# Run unit tests for Python 3.13
+env_vars: {
+    key: "NOX_SESSION"
+    value: "unit-3.13"
+}
+
+# Run unit tests in parallel, splitting up by file
+env_vars: {
+    key: "PYTEST_ADDOPTS"
+    value: "-n=auto --dist=loadscope"
+}
diff --git a/CONTRIBUTING.rst b/CONTRIBUTING.rst
index 558af9f5e4..ff71365e69 100644
--- a/CONTRIBUTING.rst
+++ b/CONTRIBUTING.rst
@@ -22,7 +22,7 @@ In order to add a feature:
   documentation.
 
 - The feature must work fully on the following CPython versions:
-  3.9, 3.10, 3.11 and 3.12 on both UNIX and Windows.
+  3.9, 3.10, 3.11, 3.12, and 3.13 on both UNIX and Windows.
 
 - The feature must not add unnecessary dependencies (where
   "unnecessary" is of course subjective, but new dependencies should
@@ -225,11 +225,13 @@ We support:
 -  `Python 3.10`_
 -  `Python 3.11`_
 -  `Python 3.12`_
+-  `Python 3.13`_
 
 .. _Python 3.9: https://docs.python.org/3.9/
 .. _Python 3.10: https://docs.python.org/3.10/
 .. _Python 3.11: https://docs.python.org/3.11/
 .. _Python 3.12: https://docs.python.org/3.12/
+.. _Python 3.13: https://docs.python.org/3.13/
 
 
 Supported versions can be found in our ``noxfile.py`` `config`_.
diff --git a/setup.py b/setup.py
index 83041fd847..b973f6935f 100644
--- a/setup.py
+++ b/setup.py
@@ -93,7 +93,10 @@
     "requests >= 2.28.1",
 ]
 
-autologging_extra_require = ["mlflow>=1.27.0,<=2.16.0"]
+autologging_extra_require = [
+    "mlflow>=1.27.0,<=2.16.0; python_version<'3.13'",
+    "mlflow>=1.27.0; python_version>='3.13'",
+]
 
 preview_extra_require = []
 
diff --git a/testing/constraints-3.13.txt b/testing/constraints-3.13.txt
index 110ffaed08..a1fae03754 100644
--- a/testing/constraints-3.13.txt
+++ b/testing/constraints-3.13.txt
@@ -12,4 +12,5 @@ ray==2.5.0 # Pinned until 2.9.3 is verified for Ray tests
 ipython==8.22.2 # Pinned to unbreak TypeAliasType import error
 google-adk==0.0.2
 google-genai>=1.10.0
-google-vizier==0.1.21
\ No newline at end of file
+google-vizier==0.1.21
+pyarrow>=18.0.0
\ No newline at end of file
diff --git a/tests/unit/vertexai/test_evaluation.py b/tests/unit/vertexai/test_evaluation.py
index 3c7b7fd270..747285984a 100644
--- a/tests/unit/vertexai/test_evaluation.py
+++ b/tests/unit/vertexai/test_evaluation.py
@@ -16,6 +16,7 @@
 #
 
 import re
+import sys
 import threading
 import time
 from unittest import mock
@@ -1061,6 +1062,9 @@ def test_compute_pointwise_metrics_without_model_inference(self, api_transport):
             "explanation",
         ]
 
+    @pytest.mark.skipif(
+        sys.version_info >= (3, 13), reason="flaky race condition in python 3.13"
+    )
     @pytest.mark.parametrize("api_transport", ["grpc", "rest"])
     def test_compute_model_based_translation_metrics_without_model_inference(
         self, api_transport
@@ -1372,6 +1376,9 @@ def test_compute_pairwise_metrics_without_model_inference(self, api_transport):
             == 0.5
         )
 
+    @pytest.mark.skipif(
+        sys.version_info >= (3, 13), reason="flaky race condition in python 3.13"
+    )
     @pytest.mark.parametrize("api_transport", ["grpc", "rest"])
     def test_compute_multiple_metrics(self, api_transport):
         aiplatform.init(
@@ -1499,6 +1506,9 @@ def test_eval_result_experiment_run_logging(self):
             {"row_count": 1, "mock_metric/mean": 1.0, "mock_metric/std": "NaN"}
         )
 
+    @pytest.mark.skipif(
+        sys.version_info >= (3, 13), reason="flaky race condition in python 3.13"
+    )
     @pytest.mark.parametrize("api_transport", ["grpc", "rest"])
     def test_rubric_based_instruction_following_metric(self, api_transport):
         aiplatform.init(