multifield-adaptive-retrieval/pyproject.toml at main · microsoft/multifield-adaptive-retrieval · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
[tool.poetry]
name = "mfar"
version = "0.1.0"
description = ""
authors = ["Patrick Xia <[email protected]>"]
readme = "README.md"

[tool.poetry.dependencies]
python = ">=3.10,<3.12"
torch = "^2.5.1"  # torch 1.13 has cuda linking issues when custom cuda is present on AzureML machines
tqdm = "^4.65.0"
transformers = ">4.27.4" # This was exactly 4.27.4 because of mlflow related issues, check back on it later?
datasets = "^2.11.0"
xmltodict = "^0.13.0"
mwparserfromhell = "^0.6.4"
fire = "^0.5.0"
beir = "^1.0.1"
gdown = "^4.7.1"
mashumaro = "^3.7"
ipython = "^8.12.0"
pytorch-lightning = "^2.0.2"
more-itertools = "^9.1.0"
tensorboardX = "^2.6.0"
pyarrow = "=15.0.0"
jsonlines = "^4.0.0"
bm25s = {extras = ["full"], version = "^0.1.10"}
wandb = "^0.17.5"
stark-qa = "^0.1.3"
pystemmer = "^2.2.0.1"
jax = {extras = ["cpu"], version = "^0.4.31"}
mlflow-skinny = "^2.15.1"
azureml-mlflow = "^1.57.0"


[[tool.poetry.source]]
name = "aiinfra"
url = "https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/"
priority = "explicit"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"