# -------- dl backend -------- #
torch==2.0.0
pytorch-lightning==2.1.2

# --------- pytorch-ie --------- #
pytorch-ie>=0.31.4,<0.32.0
pie-datasets>=0.10.5,<0.11.0
pie-modules>=0.14.2,<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.5.0
arxiv~=2.1.3
# data preparation
acl-anthology-py>=0.4.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)