requirements.txt

# --------- pytorch-ie --------- #
pytorch-ie>=0.23.0,<1.0.0
# pie-utils provides some useful helper methods for pytorch-ie,
# e.g. document processors or span utils (convert span annotations
# to sequence encodings such as BIO, IO or BIOUL, and back).
git+https://github.com/ArneBinder/pie-utils@v0.5.1

# --------- 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
wandb