requirements.txt

# --------- pytorch-ie --------- #
pytorch-ie>=0.29.6,<0.32.0
pie-datasets>=0.8.2,<0.11.0
pie-modules>=0.12.1,<0.13.0

# --------- hydra --------- #
hydra-core>=1.3.0
hydra-colorlog>=1.2.0
hydra-optuna-sweeper>=1.2.0

# --------- loggers --------- #
# wandb
# neptune-client
# mlflow
# comet-ml
# tensorboard
# aim

# --------- 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 --------- #
pyrootutils     # standardizing the project root setup
python-dotenv   # loading env variables from .env file
rich            # beautiful text formatting in terminal
pytest          # tests
pytest-cov      # test coverage
sh              # for running bash commands in some tests
pudb            # debugger
tabulate        # show statistics as markdown
plotext         # show statistics as plots
# huggingface-hub>=0.13  # interaction with HF hub