-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathpyproject.toml
73 lines (63 loc) · 2.02 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
64
65
66
67
68
69
70
71
72
[build-system]
requires = ["poetry-core", "setuptools", "wheel"]
build-backend = "poetry.core.masonry.api"
[tool.poetry]
name = "zero-shot-replication"
version = "0.1.0"
description = "Zero-Shot Replication Framework"
authors = ["Owen Colegrove <[email protected]>"]
license = "Apache-2.0"
readme = "README.md"
[tool.poetry.dependencies]
python = ">=3.11,<3.12"
accelerate = "^0.22.0"
anthropic = "^0.3.10"
astunparse = "1.6.3"
bitsandbytes = { version = "^0.41.1", optional = true }
black = "^23.3.0"
evalplus = "^0.1.6"
numpy = "^1.25.2"
openai = "0.27.8"
pandas = "^2.0.3"
python-dotenv = "^1.0.0"
python-leetcode = { version = "1.2.1", optional = true }
sentencepiece = "^0.1.99"
torch = "2.0.0"
protobuf = "~4.23.4"
vllm = { version = "0.1.4", optional = true }
scipy = { version = "^1.11.2", optional = true }
# Remote repositories
automata = { git = "https://github.com/emrgnt-cmplxty/automata.git", branch = "feature/automata-slim", optional = true }
leetcode_env = { path = "zero_shot_replication/evals/leetcode_hard_gym", optional = true }
# Hand managed dependencies
# For use with all non-Phind models
transformers = "4.32.0"
# For use with `Phind` model
# transformers = { git = "https://github.com/huggingface/transformers.git", branch = "main" }
[tool.poetry.extras]
vllm_support = ["vllm"]
# For running `automata` evaluation
automata_support = ["automata"]
# For running `leetcode` evaluation
leetcode_support = ["python-leetcode", "leetcode_env"]
# For running `HumanEval` and `HumanEval+` evaluations
humaneval_support = ["evalplus"]
# For running quantized models
quantized_support = ["bitsandbytes", "scipy"]
[tool.poetry.group.dev.dependencies]
flake8 = "6.1.0"
isort = "5.12.0"
pre-commit = "^3.3.3"
mypy = "^1.5.1"
sourcery = "^1.6.0"
types-requests = "^2.31.0.2"
types-attrs = "^19.1.0"
yapf = "0.40.1"
[tool.black]
line-length = 79
[tool.mypy]
ignore_missing_imports = true
exclude = 'zero_shot_replication/evals/leetcode_hard_gym'
[tool.flake8]
ignore = ["E501", "W503"]
exclude = 'zero_shot_replication/evals/leetcode_hard_gym'