numpy<1.29,>=1.21
scipy<1.13,>=1.5.4
pandas<2.2.0,>=2.0.0
scikit-learn<1.5,>=1.3.0
Pillow<11,>=10.0.1
tqdm<5,>=4.38
boto3<2,>=1.10
torch<2.1,>=2.0
lightning<2.1,>=2.0.0
requests<3,>=2.21
jsonschema<4.18,>=4.14
seqeval<1.3.0,>=1.2.2
evaluate<0.5.0,>=0.4.0
accelerate<0.22.0,>=0.21.0
transformers[sentencepiece]<4.32.0,>=4.31.0
timm<0.10.0,>=0.9.5
torchvision<0.16.0,>=0.14.0
scikit-image<0.21.0,>=0.19.1
text-unidecode<1.4,>=1.3
torchmetrics<1.2.0,>=1.0.0
nptyping<2.5.0,>=1.4.4
omegaconf<2.3.0,>=2.1.1
autogluon.core[raytune]==0.8.3b20231116
autogluon.features==0.8.3b20231116
autogluon.common==0.8.3b20231116
pytorch-metric-learning<2.0,>=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

[PyMuPDF]
PyMuPDF<=1.21.1

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

[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.5.4,>=8.5.3.1
