| # -------- dl backend -------- # | |
| torch==2.0.0 | |
| pytorch-lightning==2.1.2 | |
| # --------- pytorch-ie --------- # | |
| pytorch-ie>=0.31.8,<0.32.0 | |
| pie-datasets>=0.10.10,<0.11.0 | |
| pie-modules>=0.15.6,<0.16.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.5.0 | |
| arxiv~=2.1.3 | |
| # data preparation | |
| acl-anthology-py>=0.4.3 | |
| # see https://github.com/gradio-app/gradio/issues/10662 | |
| pydantic==2.10.6 | |
| # --------- 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) | |