-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
70 lines (62 loc) · 1.32 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
[build-system]
requires = ["setuptools"]
[project]
name = "influence_benchmark"
version = "0.0.1"
description = "Influence-Benchmark"
authors = [{ name = "Marcus Williams", email = "[email protected]" }]
# Dynamically include the README file, as specified in the `tool.setuptools.dynamic` section.
dynamic = ["readme"]
dependencies = [
"numpy",
"openai",
"pyyaml",
"torch",
"transformers",
"trl",
"tqdm",
"accelerate",
"peft",
"datasets",
"matplotlib",
"pandas",
"wandb",
"flask",
"flask-socketio"
]
[tool.setuptools.dynamic]
readme = { file = ["README.md"], content-type = "text/markdown" }
[project.optional-dependencies]
dev = [
"black",
"flake8",
"pep8-naming",
"mypy",
"pytest",
"pytest-timeout",
"isort",
]
[tool.black]
line-length = 120
include = '\.pyi?$' # Includes .pyi stub files
exclude = '''
/(
tmp
)
'''
[tool.isort]
profile = "black"
line_length = 120
src_paths = ["influence_benchmark"]
known_first_party = ["influence_benchmark"]
known_third_party = ["ray", "torch", "wandb"]
skip = ["experiments/test.py", "tmp"]
# Should look into the below more
[tool.pytest.ini_options]
timeout = 60
testpaths = ["tests"]
markers = []
[tool.setuptools.packages.find]
where = ["."]
include = ["influence_benchmark*"]
exclude = ["experiments*"]