Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
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
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ hydra-core>=1.3.0
omegaconf>=2.2,<2.3
pyyaml==6.0.1
torchmetrics
transformers==4.55.4
transformers==4.57.1
wandb
webdataset>=0.1.48,<=0.1.62
pandas
Expand All @@ -29,12 +29,12 @@ pybind11
pytorch-lightning==2.5.0
ipadic
mecab-python3
protobuf==3.20.3
protobuf==4.25.8
datasets==4.1.1
dill==0.3.8
nemo_toolkit==2.1.0
regex
requests==2.32.4
requests==2.32.5
python-daemon
huggingface_hub>=0.27.1
multiprocess==0.70.16
Expand Down
4 changes: 2 additions & 2 deletions test/sagemaker_tests/autogluon/inference/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
boto3
protobuf>=3.20,<3.21
protobuf>=3.20,<6.34
sagemaker>=2.187,<3
coverage
flake8==3.7.7
Flask==1.1.1
Flask==2.2.5
mock
pytest
pytest-cov
Expand Down
4 changes: 2 additions & 2 deletions test/sagemaker_tests/autogluon/training/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
boto3
protobuf>=3.20,<3.21
protobuf>=3.20,<6.34
sagemaker>=2.187,<3
coverage
flake8==3.7.7
Flask==1.1.1
Flask==2.2.5
mock
pytest
pytest-cov
Expand Down
6 changes: 3 additions & 3 deletions test/sagemaker_tests/huggingface/inference/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -3,17 +3,17 @@ coverage
# Docker v7.0.0 breaks compatibility with Docker Compose v1 (SageMaker Local)
docker>=5,<=6.1.3
flake8==3.7.7
Flask==1.1.1
Flask==2.2.5
mock
pytest==8.3.5
pytest-cov
pytest-rerunfailures
pytest-xdist
PyYAML
protobuf>=3.20,<=3.20.2
protobuf>=3.20,<=6.33.0
sagemaker>=2.237.0,<3
six
requests<2.32.0
requests<2.33.0
requests_mock
Pillow
retrying==1.3.3
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,13 +4,13 @@ s3transfer<0.7.0,>=0.6.0
sagemaker>=2.31,<3
coverage
flake8==3.7.7
Flask==1.1.1
Flask==2.2.5
mock
pytest==8.3.5
pytest-cov
pytest-rerunfailures
pytest-xdist
requests==2.31.0
requests==2.32.4
pluggy<2,>=0.3.0
requests_mock
fabric
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ python-dateutil>=2.1,<2.8.1
requests-mock
# test pipemode fails for 2.x containers with module 'tensorflow' has no attribute 'train'
# without the below package
tensorflow==2.11.1
tensorflow==2.12.1
fabric
invoke
retrying==1.3.3
Expand Down
2 changes: 1 addition & 1 deletion test/sagemaker_tests/mxnet/inference/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ pytest==5.3.5
pytest-cov
pytest-rerunfailures==9.0
pytest-xdist
requests==2.32.2
requests==2.32.4
urllib3>=1.26.6
mock
sagemaker
Expand Down
2 changes: 1 addition & 1 deletion test/sagemaker_tests/mxnet/training/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ pytest==5.3.5
pytest-cov
pytest-rerunfailures==9.0
pytest-xdist
requests==2.25.1
requests==2.32.4
urllib3>=1.26.6
mock
sagemaker
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ coverage
# Docker v7.0.0 breaks compatibility with Docker Compose v1 (SageMaker Local)
docker==6.1.3
docker-compose
Flask==1.1.1
Flask==2.2.5
fabric
flake8==3.7.7
gitpython
Expand All @@ -19,12 +19,12 @@ pytest<8.1
pytest-cov
pytest-rerunfailures
pytest-xdist
requests<2.32.0
requests<2.33.0
requests_mock
retrying==1.3.3
sagemaker>=2,<3
sagemaker-inference
tenacity
toml
torch<2.5.0
torch<2.10.0
torchvision
6 changes: 3 additions & 3 deletions test/sagemaker_tests/pytorch/inference/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
boto3
coverage
Flask==1.1.1
Flask==2.2.5
fabric
flake8==3.7.7
gitpython
Expand All @@ -14,12 +14,12 @@ pytest<8.1
pytest-cov
pytest-rerunfailures
pytest-xdist
requests<2.32.0
requests<2.33.0
requests_mock
retrying==1.3.3
sagemaker>=2,<3
sagemaker-inference
tenacity
toml
torch<2.5.0
torch<2.10.0
torchvision
2 changes: 1 addition & 1 deletion test/sagemaker_tests/pytorch/training/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ protobuf
sagemaker>=2.180.0
coverage
flake8==3.7.7
Flask==1.1.1
Flask==2.2.5
mock==2.0.0
# Preserve pytest caching behavior in earlier versions
pytest<8.1
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,6 @@ datasets
numpy
scipy
torchnet
transformers==4.21.0
transformers==4.53.0
smdebug
humanize
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ pytest==5.3.5
pytest-cov
pytest-rerunfailures==9.0
pytest-xdist
requests==2.24
requests==2.32.4
mock
sagemaker>=2,<3
boto3
Expand Down