Closes tensorflow#7743. == Packaging == - Bump requires-python upper bound from <3.11 to <3.12 in pyproject.toml, package_build/tfx/pyproject.toml, and package_build/ml-pipelines-sdk/pyproject.toml. - Add Programming Language :: Python :: 3.11 classifier in each. == CI matrix == - Add '3.11' to ci-test.yml matrix and 'cp311' to wheels.yml matrix. - Disable strategy.fail-fast so individual job failures don't cancel the whole matrix (useful regardless of this change). == Dependency pins (test_constraints.txt + nightly_test_constraints.txt) == TFX 1.17 cp311 wheels require newer versions of several deps than the cp39/cp310 wheels. Use PEP 508 environment markers to pin per-Python-version: - apache-beam: 2.50.0 on <3.11, 2.53.0 on >=3.11 (tfx-bsl/tfdv cp311 wheels require beam >=2.53) - protobuf: 4.21.12 on <3.11, 4.25.5 on >=3.11 (tfx-bsl/tfdv/ml-metadata cp311 wheels require protobuf >=4.25.2; beam 2.53 requires !=4.21.*) - pyarrow: pin to 10.0.1 on >=3.11 (10.0.0 has no cp311 wheel and fails to build from sdist under setuptools 70) - tensorflow-io-gcs-filesystem: 0.24.0 on <3.11, 0.37.1 on >=3.11 (0.24.0 has no cp311 wheel) - grpcio-status: 1.62.3 (was 1.48.2); google-api-core requires grpcio-status>=1.49.1 on >=3.11, while grpcio-status 1.63+ would require protobuf 5 which conflicts with TFX's protobuf<5 == Test fixes == Two test files had patterns that broke on Python 3.11: - tfx/components/infra_validator/request_builder_test.py: mock.patch.object(..., wraps=Cls) returns _SentinelObject for .return_value before the wrapped class is called. Switch to autospec=True. autospec is strict about kwargs so pass examples= by keyword. - tfx/components/trainer/rewriting/tflite_rewriter_test.py: mock.patch('tensorflow.lite.TFLiteConverter.from_saved_model') fails on 3.11 because mock's string-path resolution doesn't trigger TF's lazy tf.lite module wrapper. Switch to mock.patch.object( tf.lite.TFLiteConverter, ...) for direct object reference. All 12 CI jobs (3 Python versions x not-e2e/e2e x DEFAULT/NIGHTLY) pass on this branch.

Commits

Commits on May 18, 2026

Add support for Python 3.11