diff --git a/.kokoro/samples/python3.10/common.cfg b/.kokoro/samples/python3.10/common.cfg
index a49138fd0a..17abca148a 100644
--- a/.kokoro/samples/python3.10/common.cfg
+++ b/.kokoro/samples/python3.10/common.cfg
@@ -14,7 +14,7 @@ env_vars: {
}
# Declare build specific Cloud project.
-env_vars: {
+env_vars: {
key: "BUILD_SPECIFIC_GCLOUD_PROJECT"
value: "ucaip-sample-tests"
}
diff --git a/.kokoro/samples/python3.11/common.cfg b/.kokoro/samples/python3.11/common.cfg
index c870d5b2c7..1166f2c317 100644
--- a/.kokoro/samples/python3.11/common.cfg
+++ b/.kokoro/samples/python3.11/common.cfg
@@ -14,7 +14,7 @@ env_vars: {
}
# Declare build specific Cloud project.
-env_vars: {
+env_vars: {
key: "BUILD_SPECIFIC_GCLOUD_PROJECT"
value: "ucaip-sample-tests"
}
diff --git a/.kokoro/samples/python3.7/common.cfg b/.kokoro/samples/python3.7/common.cfg
index cc8296c89d..eed23ad9bc 100644
--- a/.kokoro/samples/python3.7/common.cfg
+++ b/.kokoro/samples/python3.7/common.cfg
@@ -14,7 +14,7 @@ env_vars: {
}
# Declare build specific Cloud project.
-env_vars: {
+env_vars: {
key: "BUILD_SPECIFIC_GCLOUD_PROJECT"
value: "ucaip-sample-tests"
}
diff --git a/.kokoro/samples/python3.8/common.cfg b/.kokoro/samples/python3.8/common.cfg
index a118253a82..26e513823e 100644
--- a/.kokoro/samples/python3.8/common.cfg
+++ b/.kokoro/samples/python3.8/common.cfg
@@ -14,7 +14,7 @@ env_vars: {
}
# Declare build specific Cloud project.
-env_vars: {
+env_vars: {
key: "BUILD_SPECIFIC_GCLOUD_PROJECT"
value: "ucaip-sample-tests"
}
diff --git a/.kokoro/samples/python3.9/common.cfg b/.kokoro/samples/python3.9/common.cfg
index 5a549c80fc..abda08ed27 100644
--- a/.kokoro/samples/python3.9/common.cfg
+++ b/.kokoro/samples/python3.9/common.cfg
@@ -14,7 +14,7 @@ env_vars: {
}
# Declare build specific Cloud project.
-env_vars: {
+env_vars: {
key: "BUILD_SPECIFIC_GCLOUD_PROJECT"
value: "ucaip-sample-tests"
}
diff --git a/CONTRIBUTING.rst b/CONTRIBUTING.rst
index 953a2fc715..4334f9fb0c 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.7, 3.8, 3.9 and 3.10 on both UNIX and Windows.
+ 3.7, 3.8, 3.9, 3.10 and 3.11 on both UNIX and Windows.
- The feature must not add unnecessary dependencies (where
"unnecessary" is of course subjective, but new dependencies should
@@ -72,7 +72,7 @@ We use `nox `__ to instrument our tests.
- To run a single unit test::
- $ nox -s unit-3.10 -- -k
+ $ nox -s unit-3.11 -- -k
.. note::
@@ -225,11 +225,13 @@ We support:
- `Python 3.8`_
- `Python 3.9`_
- `Python 3.10`_
+- `Python 3.11`_
.. _Python 3.7: https://docs.python.org/3.7/
.. _Python 3.8: https://docs.python.org/3.8/
.. _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/
Supported versions can be found in our ``noxfile.py`` `config`_.
diff --git a/noxfile.py b/noxfile.py
index 87ec893504..bc9dbabd5e 100644
--- a/noxfile.py
+++ b/noxfile.py
@@ -32,7 +32,7 @@
DEFAULT_PYTHON_VERSION = "3.8"
-UNIT_TEST_PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"]
+UNIT_TEST_PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10", "3.11"]
UNIT_TEST_STANDARD_DEPENDENCIES = [
"mock",
"asyncmock",
diff --git a/owlbot.py b/owlbot.py
index 5f173be611..263dc6ee39 100644
--- a/owlbot.py
+++ b/owlbot.py
@@ -88,14 +88,14 @@
# only run post processor when there are changes to the generated code
if has_generator_updates:
-# ----------------------------------------------------------------------------
-# Add templated files
-# ----------------------------------------------------------------------------
+ # ----------------------------------------------------------------------------
+ # Add templated files
+ # ----------------------------------------------------------------------------
templated_files = common.py_library(
cov_level=98,
system_test_python_versions=["3.8"],
- unit_test_python_versions=["3.7", "3.8", "3.9", "3.10"],
+ unit_test_python_versions=["3.7", "3.8", "3.9", "3.10", "3.11"],
unit_test_extras=["testing"],
system_test_extras=["testing"],
microgenerator=True,
@@ -130,13 +130,13 @@
s.replace(
".kokoro/samples/python3.*/common.cfg",
"""env_vars: \{
- key: "BUILD_SPECIFIC_GCLOUD_PROJECT"
- value: "python-docs-samples-tests-.*?"
-\}""",
- """env_vars: {
- key: "BUILD_SPECIFIC_GCLOUD_PROJECT"
- value: "ucaip-sample-tests"
-}""",
+ key: "BUILD_SPECIFIC_GCLOUD_PROJECT"
+ value: "python-docs-samples-tests-.*?"
+ \}""",
+ """env_vars: {
+ key: "BUILD_SPECIFIC_GCLOUD_PROJECT"
+ value: "ucaip-sample-tests"
+ }""",
)
s.replace(
diff --git a/setup.py b/setup.py
index e5a1dae6ab..0b16e349a0 100644
--- a/setup.py
+++ b/setup.py
@@ -59,14 +59,16 @@
"pyarrow >= 6.0.1",
]
pipelines_extra_require = [
- "pyyaml>=5.3,<7",
+ "pyyaml==5.3.1",
]
datasets_extra_require = [
- "pyarrow >= 3.0.0, < 8.0dev",
+ "pyarrow >= 3.0.0, < 8.0dev; python_version<'3.11'",
+ "pyarrow >= 10.0.1; python_version>='3.11'",
]
vizier_extra_require = [
- "google-vizier==0.0.4",
+ "google-vizier==0.0.4; python_version<'3.11'",
+ "google-vizier>=0.1.6; python_version>='3.11'",
]
prediction_extra_require = [
diff --git a/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf2_test.py b/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf2_test.py
index a6685c5285..3f6be1dc3d 100644
--- a/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf2_test.py
+++ b/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf2_test.py
@@ -116,6 +116,7 @@ def call(self, inputs):
}
assert expected_md == generated_md
+ @pytest.mark.skip(reason="Failing for Python 3.11, tracked in b/293506827.")
def test_non_keras_model(self):
class CustomModuleWithOutputName(tf.Module):
def __init__(self):