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-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-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.
# 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.
# For IFEval dataset loading in kvbm tests
datasets
datasets==4.4.1
# For NATS object store verification in router tests
nats-py
psutil>=5.0.0
pyright
pytest
pytest-asyncio
pytest-benchmark
pytest-codeblocks
pytest-cov
pytest-forked
pytest-httpserver
pytest-md-report
pytest-mypy
pytest-timeout
pytest-xdist
nats-py==2.12.0
psutil<=7.0.0 # System package, may vary by platform (was >=5.0.0)
pyright==1.1.407
pytest==8.4.2
pytest-asyncio==1.3.0
pytest-benchmark==5.2.3
pytest-codeblocks==0.17.0
pytest-cov==7.0.0
pytest-forked==1.6.0
pytest-httpserver==1.1.3
pytest-md-report==0.7.0
pytest-mypy==1.0.1
pytest-timeout==2.4.0
pytest-xdist==3.8.0
# Triton client to Dynamo gRPC server
tritonclient[grpc]
tritonclient[grpc]<=2.62.0 # May have platform-specific builds
# add types library stub for PyYAML
types-PyYAML
types-requests
types-PyYAML==6.0.12.20250915
types-requests==2.32.4.20250913
# SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# 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)
accelerate
accelerate==1.12.0
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
av==15.0.0
fastapi==0.120.1
ftfy
ftfy==6.3.1
genai-perf==0.0.15
grpcio-tools==1.66.0
httpx
kr8s
grpcio-tools<=1.66.0 # May have platform-specific builds
httpx==0.28.1
kr8s==0.20.13
kubernetes==32.0.1
kubernetes_asyncio
matplotlib
msgspec
mypy
nvidia-ml-py==13.580.65
opentelemetry-api
opentelemetry-sdk
pip
pmdarima
pre-commit
prometheus-api-client
prometheus_client
prophet
kubernetes_asyncio<=32.1.1 # May vary by platform
matplotlib==3.10.7
msgspec==0.19.0
mypy==1.18.2
nvidia-ml-py<=13.580.65 # NVIDIA/CUDA related, may vary by driver version
opentelemetry-api<=1.38.0 # May need to stay in sync with other components
opentelemetry-sdk<=1.38.0 # May need to stay in sync with other components
pip<=25.0.1 # System pip, varies by platform
pmdarima==2.1.1
pre-commit==4.5.0
prometheus-api-client==0.6.0
prometheus_client==0.23.1
prophet==1.2.1
protobuf==5.29.5
pydantic>=2.10.6
pyright
PyYAML
scikit-learn
scipy<1.14.0 # Pin scipy version for pmdarima compatibility
sentencepiece
pydantic>=2.11.4,<2.12 # Required by aiconfigurator==0.4.0
pyright==1.1.407
PyYAML==6.0.3
scikit-learn==1.7.2
scipy<1.14.0 # Upper bound for pmdarima compatibility
sentencepiece==0.2.1
tensorboard==2.19.0
tensorboardX==2.6.2.2
# Transformers version constraint for container builds
# - vLLM 0.11.0: >=4.55.2, vLLM 0.11.2: >=4.56.0,<5
# - TensorRT-LLM 1.2.0rc2/rc3: ==4.56.0
# - 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
types-aiofiles
types-PyYAML
uvicorn
types-aiofiles==25.1.0.20251011
types-PyYAML==6.0.12.20250915
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