requirements.txt

# --------- pytorch --------- #
torch>=1.10.0
torchvision>=0.11.0
pytorch-lightning>=1.5.10
torchmetrics>=0.7.0

# --------- hydra --------- #
hydra-core>=1.1.0
hydra-colorlog>=1.1.0
hydra-optuna-sweeper>=1.1.0
omegaconf>=2.1.1

# --------- loggers --------- #
wandb==0.11.2

# --------- linters --------- #
pre-commit      # hooks for applying linters on commit
black           # code formatting
isort           # import sorting
flake8          # code analysis
nbstripout      # remove output from jupyter notebooks

# --------- others --------- #
python-dotenv   # loading env variables from .env file
rich            # beautiful text formatting in terminal
pytest          # tests
sh              # for running bash commands in some tests
pudb            # debugger

# --------- dataframes --------- #
pandas>=1.4.2           # dataframes
openpyxl>=3.0.9         # xlsx support

# --------- imputation --------- #
impyute>=0.0.8          # imputation methods

# --------- plot --------- #
plotly>=5.7.0
matplotlib>=3.5.2
seaborn>=0.11.2
UpSetPlot>=0.6.1
kaleido==0.1.0post1

# --------- XAI --------- #
shap==0.39.0            # SHAP values
lime>=0.2.0.1           # LIME weights

# --------- models --------- #
pytorch-tabnet>=3.1.1
xgboost>=1.6.0
catboost>=1.0.5
lightgbm>=3.3.2
statsmodels>=0.13.1
scikit-learn>=1.0.2
einops>=0.4.1
numpy>=1.22.3

# --------- utils --------- #
tqdm>=4.63.1