NumPy>=1.16.2
lightgbm>=2.3.1
xgboost>=0.90
scipy>=1.4.1
catboost>=0.23
scikit-learn>=0.23.2

[azureml]
azureml-mlflow

[blendsearch]
optuna==2.8.0

[nlp]
ray[tune]>=1.4.1
transformers
datasets==1.4.1
tensorboardX<=2.2
torch

[nni]
nni

[notebook]
openml==0.10.2
jupyter
matplotlib==3.2.0
rgf-python
vowpalwabbit

[ray]
ray[tune]==1.4.1
pyyaml<5.3.1

[test]
flake8>=3.8.4
pytest>=6.1.1
coverage>=5.3
xgboost<1.3
rgf-python
optuna==2.8.0
vowpalwabbit
openml
transformers==4.4.1
wandb==0.10.26
torch==1.8.1
datasets==1.4.1
azure-storage-blob

[vw]
vowpalwabbit
