dispatchqa/pyproject.toml at master · upwork/dispatchqa · 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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
[project]
name = "dispatchqa"
version = "0.1.0"
description = "DispatchQA project dependencies"
requires-python = ">=3.10"
dependencies = [
    "openai==1.78.1",
    "tqdm==4.67.1",
    "transformers==4.51.1",
    "accelerate==1.0.1",
    "compressed-tensors==0.11.0",
    "outlines==0.1.11",
    "beautifulsoup4==4.11.1",
    "cleantext==1.1.4",
    "ijson==3.3.0",
    "env==0.1.0",
    "Flask>=3.0.0",
    "Werkzeug>=3.0.0",
    "gdown",
    "gradio",
    "gym==0.24.0",
    "matplotlib",
    "numpy>1.22.4",
    "pandas>=2.0.0",
    "peft==0.17.1",
    "pyserini==0.42.0",
    "pytest",
    "PyYAML==6.0.1",
    "rank_bm25==0.2.2",
    "requests==2.27.1",
    "requests_mock",
    "rich==12.4.4",
    "scikit_learn>1.1.1",
    "selenium==4.2.0",
    "sentence-transformers==5.1.0",
    "thefuzz==0.19.0",
    "train==0.0.5",
    "spacy==3.7.6",
    "tenacity==8.2.3",
    "ranx==0.3.20",
    "sglang[all]==0.4.6.post5",
    "uvloop",
    "pyzmq",
    "wandb==0.21.3",
    "flashinfer-python", # Temporarily removed as it might force newer torch
]

[project.optional-dependencies]
dev = [
    "pytest",
    "black",
    "isort",
    "flake8",
]

[tool.setuptools]
packages = []  # Don't include any packages - this is just for dependencies