torch>=1.4
category-encoders==2.2.2
numpy>=1.17.2
pandas==1.1.5
scikit-learn>=0.23.2
pytorch-lightning==1.3.6
omegaconf>=2.0.1
torchmetrics>=0.3.2
tensorboard!=2.5.0,>=2.2.0
pytorch-tabnet==3.0.0
PyYAML>=5.1.*
matplotlib>3.1
ipywidgets
einops==0.3.0

[:python_version == "3.6"]
dataclasses

[all]
wandb==0.10.11
plotly==4.14.*
