numpy<1.29,>=1.21
scipy<1.13,>=1.5.4
pandas<2.3.0,>=2.0.0
scikit-learn<1.4.1,>=1.3.0
Pillow<11,>=10.0.1
tqdm<5,>=4.38
boto3<2,>=1.10
torch<2.4,>=2.2
lightning<2.4,>=2.2
transformers[sentencepiece]<4.41.0,>=4.38.0
accelerate<0.22.0,>=0.21.0
requests<3,>=2.21
jsonschema<4.22,>=4.18
seqeval<1.3.0,>=1.2.2
evaluate<0.5.0,>=0.4.0
timm<0.10.0,>=0.9.5
torchvision<0.19.0,>=0.16.0
scikit-image<0.21.0,>=0.19.1
text-unidecode<1.4,>=1.3
torchmetrics<1.3.0,>=1.2.0
nptyping<2.5.0,>=1.4.4
omegaconf<2.3.0,>=2.1.1
autogluon.core[raytune]==1.1.2b20240705
autogluon.features==1.1.2b20240705
autogluon.common==1.1.2b20240705
pytorch-metric-learning<2.4,>=1.3.0
nlpaug<1.2.0,>=1.1.10
nltk<4.0.0,>=3.4.5
openmim<0.4.0,>=0.3.7
defusedxml<0.7.2,>=0.7.1
jinja2<3.2,>=3.0.3
tensorboard<3,>=2.9
pytesseract<0.3.11,>=0.3.9
nvidia-ml-py3==7.352.0
pdf2image<1.19,>=1.17.0

[tests]
black~=23.0
isort>=5.10
datasets<2.15.0,>=2.10.0
onnx<1.14.0,>=1.13.0

[tests:platform_system != "Darwin"]
onnxruntime-gpu<1.16.0,>=1.15.0

[tests:platform_system == "Darwin"]
onnxruntime<1.16.0,>=1.15.0

[tests:platform_system == "Linux" and python_version < "3.11"]
tensorrt<8.6.2,>=8.6.0
