requirements.txt

black           # code formatting
flake8          # code analysis
hydra-colorlog>=1.1.0

# --------- hydra --------- #
hydra-core>=1.1.0
hydra-optuna-sweeper>=1.1.0
isort           # import sorting

# --------- loggers --------- #
# wandb
# neptune-client
mlflow
nbstripout      # remove output from jupyter notebooks
# comet-ml
# tensorboard

# --------- linters --------- #
pre-commit      # hooks for applying linters on commit
pudb            # debugger
pytest          # tests

# --------- others --------- #
python-dotenv   # loading env variables from .env file
pytorch-lightning==1.5.10
rich            # beautiful text formatting in terminal
sh              # for running bash commands in some tests
numpy           # for common mathematical calculations

# --------- pytorch --------- #
torch
torchmetrics>=0.7.0
torchvision
pytorch-lightning

# -------- statistics -------- #
statsmodels
pyDOE           # a library to construct experimental designs, LHS
thop
# -------- data -------------- #
h5py
monai