Skip to content

Exclude the yanked package in the requirement file and fix the curren… #3027

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
kanglant wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Exclude the yanked package in the requirement file and fix the curren… #3027

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 7 additions & 5 deletions dependencies/requirements/base_requirements/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,17 +3,19 @@ aqtp
array-record
cloud-accelerator-diagnostics
# TODO(b/477743392): Remove the pinned versions once the yanked version detection problem is resolved
cloud-tpu-diagnostics==0.1.5
cloud-tpu-diagnostics!=1.1.14
datasets
flax
# TODO: Remove flax's exclusion once the flatten-up-to errors are resolved.
flax!=0.12.3
gcsfs
google-api-python-client
google-cloud-aiplatform
google-cloud-mldiagnostics
google-cloud-monitoring
grain[parquet]
huggingface_hub
hf-xet==1.2.0
# TODO: Remove the pin once the new httpx-based request errors are fixed
huggingface_hub==0.36.0
hf-xet!=1.2.1
jax
jaxlib
jaxtyping
Expand All @@ -40,7 +42,7 @@ tensorflow-datasets
tensorflow-text
tensorflow
tiktoken
tokamax==0.0.8
tokamax!=0.1.0
transformers
qwix
google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip
Expand Down
42 changes: 25 additions & 17 deletions dependencies/requirements/generated_requirements/tpu-requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,10 @@ black>=25.12.0
blobfile>=3.1.0
build>=1.3.0
certifi>=2025.11.12
cffi>=2.0.0 ; platform_python_implementation != 'PyPy'
cfgv>=3.5.0
charset-normalizer>=3.4.4
cheroot>=11.1.2
chex>=0.1.91
click>=8.3.1
cloud-accelerator-diagnostics>=0.1.1
Expand All @@ -30,7 +32,8 @@ cloudpickle>=3.1.2
clu>=0.0.12
colorama>=0.4.6
contourpy>=1.3.3
coverage>=7.13.1
coverage>=7.13.2
cryptography>=46.0.4
cycler>=0.12.1
datasets>=4.5.0
decorator>=5.2.1
Expand All @@ -40,7 +43,8 @@ distro>=1.9.0
dm-tree>=0.1.9
docstring-parser>=0.17.0
editdistance>=0.8.1
einops>=0.8.1
einops>=0.8.2
einshape>=1.0
etils>=1.13.0
evaluate>=0.4.6
execnet>=2.1.2
Expand All @@ -57,8 +61,8 @@ google-api-core>=2.29.0
google-api-python-client>=2.188.0
google-auth-httplib2>=0.3.0
google-auth-oauthlib>=1.2.4
google-auth>=2.47.0
google-cloud-aiplatform>=1.134.0
google-auth>=2.48.0
google-cloud-aiplatform>=1.135.0
google-cloud-appengine-logging>=1.8.0
google-cloud-audit-log>=0.4.0
google-cloud-bigquery>=3.40.0
Expand All @@ -70,22 +74,21 @@ google-cloud-resource-manager>=1.16.0
google-cloud-storage>=3.8.0
google-crc32c>=1.8.0
google-genai>=1.60.0
google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip
google-pasta>=0.2.0
google-resumable-media>=2.8.0
google-tunix>=0.1.3
googleapis-common-protos>=1.72.0
grain>=0.2.15
grpc-google-iam-v1>=0.14.3
grpcio-status>=1.71.2
grpcio-status>=1.76.0
grpcio>=1.76.0
gviz-api>=1.10.0
h11>=0.16.0
h5py>=3.15.1
hf-transfer>=0.1.9
hf-xet>=1.2.0 ; platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64'
hf-xet>=1.2.0
httpcore>=1.0.9
httplib2>=0.31.1
httplib2>=0.31.2
httpx>=0.28.1
huggingface-hub>=0.36.0
humanize>=4.15.0
Expand All @@ -98,9 +101,10 @@ importlib-metadata>=8.7.1
importlib-resources>=6.5.2
iniconfig>=2.3.0
isort>=7.0.0
jaraco-functools>=4.4.0
jax>=0.8.1
jaxlib>=0.8.1
jaxtyping>=0.3.5
jaxtyping>=0.3.6
jinja2>=3.1.6
joblib>=1.5.3
jsonlines>=4.0.0
Expand All @@ -122,11 +126,11 @@ mdurl>=0.1.2
ml-collections>=1.1.0
ml-dtypes>=0.5.4
ml-goodput-measurement>=0.0.16
mlperf-logging @ https://github.com/mlcommons/logging/archive/38ab22670527888c8eb7825a4ece176fcc36a95d.zip
more-itertools>=10.8.0
mpmath>=1.3.0
msgpack>=1.1.2
msgspec>=0.20.0
multidict>=6.7.0
multidict>=6.7.1
multiprocess>=0.70.18
mypy-extensions>=1.1.0
namex>=0.1.0
Expand All @@ -149,8 +153,8 @@ orbax-checkpoint>=0.11.32
packaging>=25.0
pandas>=3.0.0
parameterized>=0.9.0
pathspec>=1.0.3
pathwaysutils>=0.1.3
pathspec>=1.0.4
pathwaysutils>=0.1.4
pillow>=12.0.0
platformdirs>=4.5.1
pluggy>=1.6.0
Expand All @@ -160,12 +164,13 @@ prometheus-client>=0.24.1
promise>=2.3
propcache>=0.4.1
proto-plus>=1.27.0
protobuf>=5.29.5
protobuf>=6.33.4
psutil>=7.1.3
pyarrow>=23.0.0
pyasn1-modules>=0.4.2
pyasn1>=0.6.2
pycnite>=2024.7.31
pycparser>=3.0 ; implementation_name != 'PyPy' and platform_python_implementation != 'PyPy'
pycryptodomex>=3.23.0
pydantic-core>=2.41.5
pydantic>=2.12.5
Expand Down Expand Up @@ -208,12 +213,12 @@ tabulate>=0.9.0
tenacity>=9.1.2
tensorboard-data-server>=0.7.2
tensorboard-plugin-profile>=2.13.0
tensorboard>=2.19.0
tensorboard>=2.20.0
tensorboardx>=2.6.4
tensorflow-datasets>=4.9.9
tensorflow-metadata>=1.17.3
tensorflow-text>=2.19.0
tensorflow>=2.19.1
tensorflow-text>=2.20.0
tensorflow>=2.20.0
tensorstore>=0.1.79
termcolor>=3.2.0
tiktoken>=0.12.0
Expand All @@ -225,6 +230,7 @@ toolz>=1.1.0
tqdm>=4.67.1
transformers>=4.57.6
treescope>=0.1.10
typeguard>=2.13.3
typing-extensions>=4.15.0
typing-inspection>=0.4.2
tzdata>=2025.3 ; sys_platform == 'emscripten' or sys_platform == 'win32'
Expand All @@ -237,6 +243,8 @@ websockets>=15.0.1
werkzeug>=3.1.3
wheel>=0.46.1
wrapt>=2.0.1
xprof>=2.21.4
xxhash>=3.6.0
yarl>=1.22.0
zipp>=3.23.0
zstandard>=0.25.0
Loading