Merge pull request #4317 from AI-Hypercomputer:fix_pip_check_dependencies

PiperOrigin-RevId: 941291334

Author: Google-ML-Automation

src/dependencies/requirements/base_requirements/requirements.txt

@@ -46,3 +46,8 @@ tokamax!=0.1.0
 transformers>=5.8.0
 uvloop
 qwix>=0.1.6
+coverage
+blobfile
+shortuuid
+nltk
+evaluate

src/dependencies/requirements/generated_requirements/cuda12-requirements.txt

@@ -3,48 +3,51 @@
 
 absl-py>=2.4.0
 aiofiles>=25.1.0
-aiohappyeyeballs>=2.6.2
-aiohttp>=3.13.5
+aiohappyeyeballs>=2.7.1
+aiohttp>=3.14.1
 aiosignal>=1.4.0
 annotated-doc>=0.0.4
 annotated-types>=0.7.0
 antlr4-python3-runtime>=4.9.3
-anyio>=4.13.0
+anyio>=4.14.1
 aqtp>=0.9.0
 array-record>=0.8.3
 astroid>=4.0.4
 astunparse>=1.6.3
 attrs>=26.1.0
 auditwheel>=6.6.0
 black>=25.12.0
+blobfile>=3.2.0
 build>=1.4.3
 certifi>=2026.2.25
 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
+chex>=0.1.92
+click>=8.4.2
 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
+coverage>=7.14.3
+cryptography>=49.0.0
 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
-drjax>=0.1.4
+drjax>=0.2.0
 editdistance>=0.8.1
 einops>=0.8.2
 einshape>=1.0
 etils>=1.14.0
+evaluate>=0.4.6
 execnet>=2.1.2
-fastapi>=0.136.1
+fastapi>=0.139.0
 filelock>=3.28.0
 flatbuffers>=25.12.19
 flax>=0.12.7
@@ -53,40 +56,40 @@ 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-auth>=2.53.0
+google-api-core>=2.31.0
+google-api-python-client>=2.198.0
+google-auth>=2.55.1
 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-bigquery>=3.41.0
+google-cloud-aiplatform>=1.159.0
+google-cloud-appengine-logging>=1.10.0
+google-cloud-audit-log>=0.6.0
+google-cloud-bigquery>=3.42.1
 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-resource-manager>=1.17.0
-google-cloud-storage>=3.10.1
-google-cloud-storage-control>=1.11.0
+google-cloud-logging>=3.16.0
+google-cloud-mldiagnostics>=1.0.4
+google-cloud-monitoring>=2.31.0
+google-cloud-resource-manager>=1.18.0
+google-cloud-storage>=3.12.0
+google-cloud-storage-control>=1.12.0
 google-crc32c>=1.8.0
-google-genai>=2.4.0
+google-genai>=2.10.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.18
 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
+httplib2>=0.32.0
 httpx>=0.28.1
-huggingface-hub>=1.15.0
-humanize>=4.15.0
+huggingface-hub>=1.21.0
+humanize>=4.16.0
 hypothesis>=6.142.1
 identify>=2.6.19
 idna>=3.11
@@ -99,14 +102,16 @@ 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.11
 jinja2>=3.1.6
+joblib>=1.5.3
 jsonlines>=4.0.0
 keras>=3.14.0
 kiwisolver>=1.5.0
 latex2sympy2-extended>=1.11.0
 libclang>=18.1.1
 libcst>=1.8.6
+lxml>=6.1.1
 markdown>=3.10.2
 markdown-it-py>=4.0.0
 markupsafe>=3.0.3
@@ -116,9 +121,9 @@ mccabe>=0.7.0
 mdurl>=0.1.2
 ml-collections>=1.1.0
 ml-dtypes>=0.5.4
-ml-goodput-measurement>=0.0.16
+ml-goodput-measurement>=0.2.0
 mpmath>=1.3.0
-msgpack>=1.1.2
+msgpack>=1.2.1
 msgspec>=0.21.1
 multidict>=6.7.1
 multiprocess>=0.70.19
@@ -127,6 +132,7 @@ namex>=0.1.0
 nest-asyncio>=1.6.0 ; sys_platform == 'win32'
 networkx>=3.6.1
 ninja>=1.13.0
+nltk>=3.9.4
 nodeenv>=1.10.0
 numpy>=2.0.2
 numpy-typing-compat>=20251206.2.0
@@ -141,12 +147,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.43.0
 opt-einsum>=3.4.0
 optax>=0.2.8
 optree>=0.19.0
@@ -156,9 +163,9 @@ packaging>=26.1
 pandas>=3.0.3
 parameterized>=0.9.0
 pathspec>=1.1.1
-pathwaysutils>=0.1.8
+pathwaysutils>=0.1.10
 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
@@ -172,41 +179,45 @@ pyasn1>=0.6.3
 pyasn1-modules>=0.4.2
 pycnite>=2024.7.31
 pycparser>=3.0 ; implementation_name != 'PyPy' and platform_python_implementation != 'PyPy'
+pycryptodomex>=3.23.0
 pydantic>=2.13.4
 pydantic-core>=2.46.4
 pydot>=4.0.1
 pyelftools>=0.32
 pyglove>=0.4.5
 pygments>=2.20.0
 pyink>=25.12.0
-pylint>=4.0.5
+pylint>=4.0.6
+pynvml>=13.0.1
+pyopenssl>=26.3.0
 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
-qwix>=0.1.6
-regex>=2026.5.9
+qwix>=0.1.8
+regex>=2026.6.28
 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
 seqio>=0.0.20
 setuptools>=82.0.1
 shellingham>=1.5.4
+shortuuid>=1.0.13
 simple-parsing>=0.1.8
 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 +227,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 +237,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.3
+transformer-engine>=2.16.1
+transformer-engine-cu12>=2.16.1
+transformer-engine-jax>=2.16.1
+transformers>=5.12.1
 treescope>=0.1.10
 typeguard>=2.13.3
 typer>=0.25.1
@@ -239,15 +250,15 @@ 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.5.1
 wadler-lindig>=0.1.7
 websockets>=16.0
 werkzeug>=3.1.8
 wheel>=0.46.3
-wrapt>=2.1.2
-xxhash>=3.7.0
+wrapt>=2.2.2
+xxhash>=3.8.0
 yarl>=1.24.2
 zipp>=3.23.1
 zstandard>=0.25.0