-
Notifications
You must be signed in to change notification settings - Fork 122
/
pyproject.toml
63 lines (53 loc) · 1.4 KB
/
pyproject.toml
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
58
59
60
61
62
63
[tool.poetry]
name = "picard"
version = "1.0.0"
description = "PICARD - Parsing Incrementally for Constrained Auto-Regressive Decoding from Language Models"
authors = ["Torsten Scholak <[email protected]>"]
license = "Apache-2.0"
readme = "README.md"
repository = "https://github.com/ElementAI/picard"
[tool.poetry.dependencies]
python = ">=3.7.1,<3.11"
cython = "^0.29.28"
six = "^1.16.0"
pandas = "==1.3.5"
numpy = "==1.21.5"
datasets = { rev = "cf47649eaed608fb7030f692020a0921e16f23c8", git = "https://github.com/huggingface/datasets" }
nltk = "^3.7"
pyyaml = "^6.0"
sentencepiece = "0.1.96"
tenacity = "^8.0.1"
rapidfuzz = "^2.0.5"
alive_progress = "^2.3.1"
sqlparse = "^0.4.2"
pillow = "9.0.1"
transformers = "^4.17.0"
deepspeed = { version = "0.5.10", optional = true }
torch = { version = "1.9.0", optional = true }
fastapi = "^0.74.1"
uvicorn = "^0.17.5"
[tool.poetry.dev-dependencies]
pytest = "^7.0.1"
pytest-check = "^1.0.4"
isort = "^5.10.1"
autopep8 = "^1.6.0"
wandb = "^0.12.11"
black = "^22.1.0"
mypy = "^0.931"
flake8 = "^4.0.1"
# codalab = "1.4.1"
[tool.poetry.extras]
torch = ["torch"]
deepspeed = ["deepspeed"]
[tool.black]
line-length = 120
[tool.isort]
line_length = 120
multi_line_output = 3
include_trailing_comma = true
use_parentheses = true
known_first_party = ["tests"]
default_section = "THIRDPARTY"
[build-system]
requires = ["poetry>=0.12"]
build-backend = "poetry.masonry.api"