Unverified Commit 30aba52c authored by Keiven C's avatar Keiven C Committed by GitHub
Browse files

chore: pin package versions in container requirements files (#4571)


Signed-off-by: default avatarKeiven Chang <keivenchang@users.noreply.github.com>
parent f33fdd52
# SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0 # SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
ucx-py-cu12 # Version Pinning Strategy:
# - Use == for packages that are pure Python and well-tested
# - Use <= or < for packages that may have platform-specific versions
# - Never use >= as it allows untested future versions that may introduce breaking changes,
# create non-reproducible builds, and cause dependency conflicts. Every installed version
# should be explicitly tested, not an unknown future release.
# CUDA-specific packages that may have platform variations
ucx-py-cu12<=0.45.0 # CUDA 12 variant, max version may differ by platform/architecture
# SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0 # SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License"); # Version Pinning Strategy:
# you may not use this file except in compliance with the License. # - Use == for packages that are pure Python and well-tested
# You may obtain a copy of the License at # - Use <= or < for packages that may have platform-specific versions
# # - Never use >= as it allows untested future versions that may introduce breaking changes,
# http://www.apache.org/licenses/LICENSE-2.0 # create non-reproducible builds, and cause dependency conflicts. Every installed version
# # should be explicitly tested, not an unknown future release.
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# For IFEval dataset loading in kvbm tests # For IFEval dataset loading in kvbm tests
datasets datasets==4.4.1
# For NATS object store verification in router tests # For NATS object store verification in router tests
nats-py nats-py==2.12.0
psutil>=5.0.0 psutil<=7.0.0 # System package, may vary by platform (was >=5.0.0)
pyright pyright==1.1.407
pytest pytest==8.4.2
pytest-asyncio pytest-asyncio==1.3.0
pytest-benchmark pytest-benchmark==5.2.3
pytest-codeblocks pytest-codeblocks==0.17.0
pytest-cov pytest-cov==7.0.0
pytest-forked pytest-forked==1.6.0
pytest-httpserver pytest-httpserver==1.1.3
pytest-md-report pytest-md-report==0.7.0
pytest-mypy pytest-mypy==1.0.1
pytest-timeout pytest-timeout==2.4.0
pytest-xdist pytest-xdist==3.8.0
# Triton client to Dynamo gRPC server # Triton client to Dynamo gRPC server
tritonclient[grpc] tritonclient[grpc]<=2.62.0 # May have platform-specific builds
# add types library stub for PyYAML # add types library stub for PyYAML
types-PyYAML types-PyYAML==6.0.12.20250915
types-requests types-requests==2.32.4.20250913
# SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0 # SPDX-License-Identifier: Apache-2.0
# Version Pinning Strategy:
# - Use == for packages that are pure Python and well-tested
# - Use <= or < for packages that may have platform-specific versions (CUDA, system packages)
# - Never use >= as it allows untested future versions that may introduce breaking changes,
# create non-reproducible builds, and cause dependency conflicts. Every installed version
# should be explicitly tested, not an unknown future release.
# - Platform differences: Some packages (especially CUDA-related) may have different
# maximum versions available on different platforms (x86_64 vs aarch64, different CUDA versions)
# For Multimodal EPD (required for device_map="auto" in vision model loading) # For Multimodal EPD (required for device_map="auto" in vision model loading)
accelerate accelerate==1.12.0
aiconfigurator @ git+https://github.com/ai-dynamo/aiconfigurator.git@5554d2eb8206738c66048bf2d72183e9bcd85759 aiconfigurator @ git+https://github.com/ai-dynamo/aiconfigurator.git@5554d2eb8206738c66048bf2d72183e9bcd85759
aiofiles aiofiles==24.1.0
aiperf @ git+https://github.com/ai-dynamo/aiperf.git@4d3fa29403c8f75da22a14f1f7b3aeb27db9288f aiperf @ git+https://github.com/ai-dynamo/aiperf.git@4d3fa29403c8f75da22a14f1f7b3aeb27db9288f
av==15.0.0 av==15.0.0
fastapi==0.120.1 fastapi==0.120.1
ftfy ftfy==6.3.1
genai-perf==0.0.15 genai-perf==0.0.15
grpcio-tools==1.66.0 grpcio-tools<=1.66.0 # May have platform-specific builds
httpx httpx==0.28.1
kr8s kr8s==0.20.13
kubernetes==32.0.1 kubernetes==32.0.1
kubernetes_asyncio kubernetes_asyncio<=32.1.1 # May vary by platform
matplotlib matplotlib==3.10.7
msgspec msgspec==0.19.0
mypy mypy==1.18.2
nvidia-ml-py==13.580.65 nvidia-ml-py<=13.580.65 # NVIDIA/CUDA related, may vary by driver version
opentelemetry-api opentelemetry-api<=1.38.0 # May need to stay in sync with other components
opentelemetry-sdk opentelemetry-sdk<=1.38.0 # May need to stay in sync with other components
pip pip<=25.0.1 # System pip, varies by platform
pmdarima pmdarima==2.1.1
pre-commit pre-commit==4.5.0
prometheus-api-client prometheus-api-client==0.6.0
prometheus_client prometheus_client==0.23.1
prophet prophet==1.2.1
protobuf==5.29.5 protobuf==5.29.5
pydantic>=2.10.6 pydantic>=2.11.4,<2.12 # Required by aiconfigurator==0.4.0
pyright pyright==1.1.407
PyYAML PyYAML==6.0.3
scikit-learn scikit-learn==1.7.2
scipy<1.14.0 # Pin scipy version for pmdarima compatibility scipy<1.14.0 # Upper bound for pmdarima compatibility
sentencepiece sentencepiece==0.2.1
tensorboard==2.19.0 tensorboard==2.19.0
tensorboardX==2.6.2.2 tensorboardX==2.6.2.2
# Transformers version constraint for container builds # Transformers version constraint for container builds
# - vLLM 0.11.0: >=4.55.2, vLLM 0.11.2: >=4.56.0,<5 # - vLLM 0.11.0: >=4.55.2, vLLM 0.11.2: >=4.56.0,<5
# - TensorRT-LLM 1.2.0rc2/rc3: ==4.56.0 # - TensorRT-LLM 1.2.0rc2/rc3: ==4.56.0
# - SGLang 0.5.4.post3: ==4.57.1 # - SGLang 0.5.4.post3: ==4.57.1
# Using >=4.56.0 to satisfy all frameworks # Using >=4.56.0 and <=4.57.1 to satisfy all frameworks
transformers>=4.56.0,<=4.57.1 transformers>=4.56.0,<=4.57.1
types-aiofiles types-aiofiles==25.1.0.20251011
types-PyYAML types-PyYAML==6.0.12.20250915
uvicorn uvicorn==0.38.0
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment