|
# --------- pytorch-ie --------- # |
|
pytorch-ie>=0.29.6,<0.32.0 |
|
pie-datasets>=0.10.5,<0.11.0 |
|
pie-modules>=0.14.0,<0.15.0 |
|
|
|
# --------- models -------- # |
|
adapters>=0.1.2,<0.2.0 |
|
pytorch-crf~=0.7.2 |
|
# --------- retriever -------- # |
|
langchain>=0.3.0,<0.4.0 |
|
langchain-core>=0.3.0,<0.4.0 |
|
langchain-community>=0.3.0,<0.4.0 |
|
# we use QDrant as vectorstore backend |
|
langchain-qdrant>=0.1.0,<0.2.0 |
|
qdrant-client>=1.12.0,<2.0.0 |
|
|
|
# --------- demo -------- # |
|
gradio~=5.4.0 |
|
arxiv~=2.1.3 |
|
|
|
# --------- 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 coverageataset |
|
sh # for running bash commands in some tests |
|
pudb # debugger |
|
tabulate # show statistics as markdown |
|
plotext # show statistics as plots |
|
prettytable # rendering annotated docs as table (demo) |
|
beautifulsoup4 # rendering annotated docs with displacy + highlighted relations (demo) |
|
# 0.26 seems to be broken when used with adapters, see https://github.com/adapter-hub/adapters/issues/748 |
|
huggingface_hub<0.26.0 # interaction with HF hub |
|
networkx~=3.2.1 # to handle segmented entities (e.g if HANDLE_PARTS_OF_SAME=True in demo) |
|
|