From 7c56d1248856346b953014674d9e50f7ac3c27eb Mon Sep 17 00:00:00 2001 From: maxtext authors Date: Wed, 10 Jun 2026 22:34:19 -0700 Subject: [PATCH] Update google-cloud-mldiagnostics to >=1.0.3 PiperOrigin-RevId: 930290224 --- .../base_requirements/requirements.txt | 2 +- .../cuda12-requirements.txt | 78 ++++++++++--------- .../tpu-requirements.txt | 72 ++++++++--------- .../requirements/requirements.txt | 2 +- 4 files changed, 79 insertions(+), 75 deletions(-) diff --git a/src/dependencies/requirements/base_requirements/requirements.txt b/src/dependencies/requirements/base_requirements/requirements.txt index 5ba8ee5093..ca674652c5 100644 --- a/src/dependencies/requirements/base_requirements/requirements.txt +++ b/src/dependencies/requirements/base_requirements/requirements.txt @@ -9,7 +9,7 @@ flax gcsfs google-api-python-client google-cloud-aiplatform -google-cloud-mldiagnostics +google-cloud-mldiagnostics>=1.0.3 google-cloud-monitoring grain[parquet] huggingface_hub>=1.14.0 diff --git a/src/dependencies/requirements/generated_requirements/cuda12-requirements.txt b/src/dependencies/requirements/generated_requirements/cuda12-requirements.txt index 4e52d9125a..1434ef3614 100644 --- a/src/dependencies/requirements/generated_requirements/cuda12-requirements.txt +++ b/src/dependencies/requirements/generated_requirements/cuda12-requirements.txt @@ -4,7 +4,7 @@ absl-py>=2.4.0 aiofiles>=25.1.0 aiohappyeyeballs>=2.6.2 -aiohttp>=3.13.5 +aiohttp>=3.14.1 aiosignal>=1.4.0 annotated-doc>=0.0.4 annotated-types>=0.7.0 @@ -23,18 +23,18 @@ cffi>=2.0.0 ; platform_python_implementation != 'PyPy' cfgv>=3.5.0 charset-normalizer>=3.4.7 chex>=0.1.91 -click>=8.4.0 +click>=8.4.1 cloud-accelerator-diagnostics>=0.1.1 cloudpickle>=3.1.2 clu>=0.0.12 colorama>=0.4.6 contourpy>=1.3.3 -cryptography>=48.0.0 +cryptography>=48.0.1 cycler>=0.12.1 -datasets>=4.8.5 +datasets>=5.0.0 decorator>=5.3.1 dill>=0.4.1 -distlib>=0.4.0 +distlib>=0.4.3 distro>=1.9.0 dm-tree>=0.1.10 docstring-parser>=0.18.0 @@ -44,7 +44,7 @@ einops>=0.8.2 einshape>=1.0 etils>=1.14.0 execnet>=2.1.2 -fastapi>=0.136.1 +fastapi>=0.136.3 filelock>=3.28.0 flatbuffers>=25.12.19 flax>=0.12.7 @@ -53,39 +53,39 @@ frozenlist>=1.8.0 fsspec>=2026.2.0 gast>=0.7.0 gcsfs>=2026.2.0 -google-api-core>=2.30.3 -google-api-python-client>=2.196.0 +google-api-core>=2.31.0 +google-api-python-client>=2.197.0 google-auth>=2.53.0 google-auth-httplib2>=0.4.0 google-auth-oauthlib>=1.4.0 -google-cloud-aiplatform>=1.153.1 -google-cloud-appengine-logging>=1.9.0 -google-cloud-audit-log>=0.5.0 +google-cloud-aiplatform>=1.157.0 +google-cloud-appengine-logging>=1.10.0 +google-cloud-audit-log>=0.6.0 google-cloud-bigquery>=3.41.0 google-cloud-core>=2.6.0 -google-cloud-logging>=3.15.0 -google-cloud-mldiagnostics>=1.0.2 -google-cloud-monitoring>=2.30.0 +google-cloud-logging>=3.16.0 +google-cloud-mldiagnostics>=1.0.3 +google-cloud-monitoring>=2.31.0 google-cloud-resource-manager>=1.17.0 -google-cloud-storage>=3.10.1 -google-cloud-storage-control>=1.11.0 +google-cloud-storage>=3.11.0 +google-cloud-storage-control>=1.12.0 google-crc32c>=1.8.0 -google-genai>=2.4.0 +google-genai>=2.8.0 google-pasta>=0.2.0 -google-resumable-media>=2.9.0 +google-resumable-media>=2.10.0 googleapis-common-protos>=1.75.0 -grain>=0.2.16 +grain>=0.2.17 grpc-google-iam-v1>=0.14.4 grpcio>=1.80.0 grpcio-status>=1.80.0 gviz-api>=1.10.0 h11>=0.16.0 h5py>=3.14.0 -hf-xet>=1.5.0 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' +hf-xet>=1.5.1 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' httpcore>=1.0.9 httplib2>=0.31.2 httpx>=0.28.1 -huggingface-hub>=1.15.0 +huggingface-hub>=1.19.0 humanize>=4.15.0 hypothesis>=6.142.1 identify>=2.6.19 @@ -99,7 +99,7 @@ jax>=0.10.0 jax-cuda12-pjrt>=0.10.0 ; sys_platform == 'linux' jax-cuda12-plugin>=0.10.0 ; sys_platform == 'linux' jaxlib>=0.10.0 -jaxtyping>=0.3.9 +jaxtyping>=0.3.10 jinja2>=3.1.6 jsonlines>=4.0.0 keras>=3.14.0 @@ -118,7 +118,7 @@ ml-collections>=1.1.0 ml-dtypes>=0.5.4 ml-goodput-measurement>=0.0.16 mpmath>=1.3.0 -msgpack>=1.1.2 +msgpack>=1.2.0 msgspec>=0.21.1 multidict>=6.7.1 multiprocess>=0.70.19 @@ -141,12 +141,13 @@ nvidia-cudnn-cu12>=9.21.0.82 ; sys_platform == 'linux' nvidia-cufft-cu12>=11.4.1.4 ; sys_platform == 'linux' nvidia-cusolver-cu12>=11.7.5.82 ; sys_platform == 'linux' nvidia-cusparse-cu12>=12.5.10.65 ; sys_platform == 'linux' +nvidia-ml-py>=13.610.43 nvidia-nccl-cu12>=2.29.7 ; sys_platform == 'linux' nvidia-nvjitlink-cu12>=12.9.86 ; sys_platform == 'linux' nvidia-nvshmem-cu12>=3.6.5 ; sys_platform == 'linux' oauthlib>=3.3.1 -omegaconf>=2.3.0 -opentelemetry-api>=1.42.0 +omegaconf>=2.3.1 +opentelemetry-api>=1.42.1 opt-einsum>=3.4.0 optax>=0.2.8 optree>=0.19.0 @@ -158,7 +159,7 @@ parameterized>=0.9.0 pathspec>=1.1.1 pathwaysutils>=0.1.8 pillow>=12.2.0 -platformdirs>=4.9.6 +platformdirs>=4.10.0 pluggy>=1.6.0 portpicker>=1.6.0 pre-commit>=4.6.0 @@ -180,12 +181,13 @@ pyglove>=0.4.5 pygments>=2.20.0 pyink>=25.12.0 pylint>=4.0.5 +pynvml>=13.0.1 pyparsing>=3.3.2 pyproject-hooks>=1.2.0 pytest>=8.4.2 pytest-xdist>=3.8.0 python-dateutil>=2.9.0.post0 -python-discovery>=1.3.1 +python-discovery>=1.4.2 pytokens>=0.4.1 pytype>=2024.10.11 pyyaml>=6.0.3 @@ -194,7 +196,7 @@ regex>=2026.5.9 requests>=2.33.1 requests-oauthlib>=2.0.0 rich>=15.0.0 -safetensors>=0.7.0 +safetensors>=0.8.0 scipy>=1.17.1 scipy-stubs>=1.17.1.4 sentencepiece>=0.2.1 @@ -206,7 +208,7 @@ simplejson>=4.1.1 six>=1.17.0 sniffio>=1.3.1 sortedcontainers>=2.4.0 -starlette>=1.0.0 +starlette>=1.3.1 sympy>=1.14.0 tabulate>=0.10.0 tenacity>=9.1.4 @@ -216,7 +218,7 @@ tensorboard-plugin-profile>=2.13.0 tensorboardx>=2.6.5 tensorflow>=2.20.0 tensorflow-datasets>=4.9.10 -tensorflow-metadata>=1.17.3 +tensorflow-metadata>=1.21.0 tensorflow-text>=2.20.1 tensorstore>=0.1.82 termcolor>=3.3.0 @@ -226,11 +228,11 @@ tokenizers>=0.22.2 toml>=0.10.2 tomlkit>=0.15.0 toolz>=1.1.0 -tqdm>=4.66.3 -transformer-engine>=2.15.0 -transformer-engine-cu12>=2.15.0 -transformer-engine-jax>=2.15.0 -transformers>=5.9.0 +tqdm>=4.68.2 +transformer-engine>=2.16.0 +transformer-engine-cu12>=2.16.0 +transformer-engine-jax>=2.16.0 +transformers>=5.11.0 treescope>=0.1.10 typeguard>=2.13.3 typer>=0.25.1 @@ -239,14 +241,14 @@ typing-inspection>=0.4.2 tzdata>=2026.2 ; sys_platform == 'emscripten' or sys_platform == 'win32' uritemplate>=4.2.0 urllib3>=2.6.3 -uvicorn>=0.47.0 +uvicorn>=0.49.0 uvloop>=0.22.1 -virtualenv>=21.3.3 +virtualenv>=21.4.3 wadler-lindig>=0.1.7 websockets>=16.0 werkzeug>=3.1.8 wheel>=0.46.3 -wrapt>=2.1.2 +wrapt>=2.2.1 xxhash>=3.7.0 yarl>=1.24.2 zipp>=3.23.1 diff --git a/src/dependencies/requirements/generated_requirements/tpu-requirements.txt b/src/dependencies/requirements/generated_requirements/tpu-requirements.txt index 26ba4fcdda..2c97272313 100644 --- a/src/dependencies/requirements/generated_requirements/tpu-requirements.txt +++ b/src/dependencies/requirements/generated_requirements/tpu-requirements.txt @@ -4,7 +4,7 @@ absl-py>=2.4.0 aiofiles>=25.1.0 aiohappyeyeballs>=2.6.2 -aiohttp>=3.13.5 +aiohttp>=3.14.1 aiosignal>=1.4.0 annotated-doc>=0.0.4 annotated-types>=0.7.0 @@ -23,18 +23,18 @@ cffi>=2.0.0 ; platform_python_implementation != 'PyPy' cfgv>=3.5.0 charset-normalizer>=3.4.7 chex>=0.1.91 -click>=8.4.0 +click>=8.4.1 cloud-accelerator-diagnostics>=0.1.1 cloudpickle>=3.1.2 clu>=0.0.12 colorama>=0.4.6 contourpy>=1.3.3 -cryptography>=48.0.0 +cryptography>=48.0.1 cycler>=0.12.1 -datasets>=4.8.5 +datasets>=5.0.0 decorator>=5.3.1 dill>=0.4.1 -distlib>=0.4.0 +distlib>=0.4.3 distro>=1.9.0 dm-tree>=0.1.10 docstring-parser>=0.18.0 @@ -44,7 +44,7 @@ einops>=0.8.2 einshape>=1.0 etils>=1.14.0 execnet>=2.1.2 -fastapi>=0.136.1 +fastapi>=0.136.3 filelock>=3.28.0 flatbuffers>=25.12.19 flax>=0.12.7 @@ -53,39 +53,39 @@ frozenlist>=1.8.0 fsspec>=2026.2.0 gast>=0.7.0 gcsfs>=2026.2.0 -google-api-core>=2.30.3 -google-api-python-client>=2.196.0 +google-api-core>=2.31.0 +google-api-python-client>=2.197.0 google-auth>=2.53.0 google-auth-httplib2>=0.4.0 google-auth-oauthlib>=1.4.0 -google-cloud-aiplatform>=1.153.1 -google-cloud-appengine-logging>=1.9.0 -google-cloud-audit-log>=0.5.0 +google-cloud-aiplatform>=1.157.0 +google-cloud-appengine-logging>=1.10.0 +google-cloud-audit-log>=0.6.0 google-cloud-bigquery>=3.41.0 google-cloud-core>=2.6.0 -google-cloud-logging>=3.15.0 -google-cloud-mldiagnostics>=1.0.2 -google-cloud-monitoring>=2.30.0 +google-cloud-logging>=3.16.0 +google-cloud-mldiagnostics>=1.0.3 +google-cloud-monitoring>=2.31.0 google-cloud-resource-manager>=1.17.0 -google-cloud-storage>=3.10.1 -google-cloud-storage-control>=1.11.0 +google-cloud-storage>=3.11.0 +google-cloud-storage-control>=1.12.0 google-crc32c>=1.8.0 -google-genai>=2.4.0 +google-genai>=2.8.0 google-pasta>=0.2.0 -google-resumable-media>=2.9.0 +google-resumable-media>=2.10.0 googleapis-common-protos>=1.75.0 -grain>=0.2.16 +grain>=0.2.17 grpc-google-iam-v1>=0.14.4 grpcio>=1.80.0 grpcio-status>=1.80.0 gviz-api>=1.10.0 h11>=0.16.0 h5py>=3.14.0 -hf-xet>=1.5.0 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' +hf-xet>=1.5.1 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' httpcore>=1.0.9 httplib2>=0.31.2 httpx>=0.28.1 -huggingface-hub>=1.15.0 +huggingface-hub>=1.19.0 humanize>=4.15.0 hypothesis>=6.142.1 identify>=2.6.19 @@ -96,7 +96,7 @@ iniconfig>=2.3.0 isort>=8.0.1 jax>=0.10.0 jaxlib>=0.10.0 -jaxtyping>=0.3.9 +jaxtyping>=0.3.10 jinja2>=3.1.6 jsonlines>=4.0.0 keras>=3.14.0 @@ -116,7 +116,7 @@ ml-collections>=1.1.0 ml-dtypes>=0.5.4 ml-goodput-measurement>=0.0.16 mpmath>=1.3.0 -msgpack>=1.1.2 +msgpack>=1.2.0 msgspec>=0.21.1 multidict>=6.7.1 multiprocess>=0.70.19 @@ -129,9 +129,10 @@ nodeenv>=1.10.0 numpy>=2.0.2 numpy-typing-compat>=20251206.2.0 nvidia-cuda-cccl>=13.2.75 +nvidia-ml-py>=13.610.43 oauthlib>=3.3.1 -omegaconf>=2.3.0 -opentelemetry-api>=1.42.0 +omegaconf>=2.3.1 +opentelemetry-api>=1.42.1 opt-einsum>=3.4.0 optax>=0.2.8 optree>=0.19.0 @@ -143,7 +144,7 @@ parameterized>=0.9.0 pathspec>=1.1.1 pathwaysutils>=0.1.8 pillow>=12.2.0 -platformdirs>=4.9.6 +platformdirs>=4.10.0 pluggy>=1.6.0 portpicker>=1.6.0 pre-commit>=4.6.0 @@ -165,12 +166,13 @@ pyglove>=0.4.5 pygments>=2.20.0 pyink>=25.12.0 pylint>=4.0.5 +pynvml>=13.0.1 pyparsing>=3.3.2 pyproject-hooks>=1.2.0 pytest>=8.4.2 pytest-xdist>=3.8.0 python-dateutil>=2.9.0.post0 -python-discovery>=1.3.1 +python-discovery>=1.4.2 pytokens>=0.4.1 pytype>=2024.10.11 pyyaml>=6.0.3 @@ -179,7 +181,7 @@ regex>=2026.5.9 requests>=2.33.1 requests-oauthlib>=2.0.0 rich>=15.0.0 -safetensors>=0.7.0 +safetensors>=0.8.0 scipy>=1.17.1 scipy-stubs>=1.17.1.4 sentencepiece>=0.2.1 @@ -191,7 +193,7 @@ simplejson>=4.1.1 six>=1.17.0 sniffio>=1.3.1 sortedcontainers>=2.4.0 -starlette>=1.0.0 +starlette>=1.3.1 sympy>=1.14.0 tabulate>=0.10.0 tenacity>=9.1.4 @@ -201,7 +203,7 @@ tensorboard-plugin-profile>=2.13.0 tensorboardx>=2.6.5 tensorflow>=2.20.0 tensorflow-datasets>=4.9.10 -tensorflow-metadata>=1.17.3 +tensorflow-metadata>=1.21.0 tensorflow-text>=2.20.1 tensorstore>=0.1.82 termcolor>=3.3.0 @@ -211,8 +213,8 @@ tokenizers>=0.22.2 toml>=0.10.2 tomlkit>=0.15.0 toolz>=1.1.0 -tqdm>=4.66.3 -transformers>=5.9.0 +tqdm>=4.68.2 +transformers>=5.11.0 treescope>=0.1.10 typeguard>=2.13.3 typer>=0.25.1 @@ -221,14 +223,14 @@ typing-inspection>=0.4.2 tzdata>=2026.2 ; sys_platform == 'emscripten' or sys_platform == 'win32' uritemplate>=4.2.0 urllib3>=2.6.3 -uvicorn>=0.47.0 +uvicorn>=0.49.0 uvloop>=0.22.1 -virtualenv>=21.3.3 +virtualenv>=21.4.3 wadler-lindig>=0.1.7 websockets>=16.0 werkzeug>=3.1.8 wheel>=0.46.3 -wrapt>=2.1.2 +wrapt>=2.2.1 xxhash>=3.7.0 yarl>=1.24.2 zipp>=3.23.1 diff --git a/src/dependencies/requirements/requirements.txt b/src/dependencies/requirements/requirements.txt index 05c2be074b..47b456c8e5 100644 --- a/src/dependencies/requirements/requirements.txt +++ b/src/dependencies/requirements/requirements.txt @@ -9,7 +9,7 @@ flax gcsfs google-api-python-client google-cloud-aiplatform -google-cloud-mldiagnostics>=0.5.10 +google-cloud-mldiagnostics>=1.0.3 google-cloud-monitoring grain[parquet] huggingface_hub