numpy<3,>=1.24
scikit-learn<2,>=1.3
pyyaml<7,>=6.0
joblib<2,>=1.3
torch<3,>=2.2
torchvision<1,>=0.17
tensorflow<3,>=2.16

[cv]
ultralytics<9,>=8.3
transformers<5,>=4.45

[dev]
build<2,>=1.2
pytest<9,>=8.2
ruff<1,>=0.5
mypy<2,>=1.10

[dl]
timm<2,>=1.0
ultralytics<9,>=8.3

[full]
xgboost<3,>=2.1
lightgbm<5,>=4.5
catboost<2,>=1.2
umap-learn<1,>=0.5
timm<2,>=1.0
ultralytics<9,>=8.3
nltk<4,>=3.9
spacy<4,>=3.7
gensim<5,>=4.3
transformers<5,>=4.45
sentence-transformers<4,>=3.0
stable-baselines3<3,>=2.3
sb3-contrib<3,>=2.3
matplotlib<4,>=3.8
seaborn<1,>=0.13

[ml_boosting]
xgboost<3,>=2.1
lightgbm<5,>=4.5
catboost<2,>=1.2
umap-learn<1,>=0.5

[nlp]
nltk<4,>=3.9
spacy<4,>=3.7
gensim<5,>=4.3
transformers<5,>=4.45
sentence-transformers<4,>=3.0

[rl]
stable-baselines3<3,>=2.3
sb3-contrib<3,>=2.3

[viz]
matplotlib<4,>=3.8
seaborn<1,>=0.13
