dvc<3,>=2.58.0
dvc-render<1.0,>=0.5.0
dvc-studio-client<1,>=0.10.0
funcy
ruamel.yaml
scmrepo

[all]
dvclive[catalyst,fastai,huggingface,image,lgbm,markdown,mmcv,optuna,plots,pytorch-lightning,tf,xgb]

[catalyst]
catalyst>22

[dev]
dvclive[all,tests]
mypy>=1.1.1

[fastai]
fastai

[huggingface]
transformers
datasets

[image]
numpy
pillow

[lgbm]
lightgbm

[lightning]
lightning>=1.9
torch<2.1

[markdown]
matplotlib

[mmcv]
mmcv

[optuna]
optuna

[plots]
dvclive[sklearn]

[pytorch_lightning]
pytorch-lightning>=1.9
torch<2.1

[sklearn]
scikit-learn

[tests]
pytest<8.0,>=7.2.0
pytest-sugar<1.0,>=0.9.6
pytest-cov<4.0,>=3.0.0
pytest-mock<4.0,>=3.8.2
dvclive[image,markdown,plots]
ipython

[tf]
tensorflow

[xgb]
xgboost
