-
Notifications
You must be signed in to change notification settings - Fork 317
/
pyproject.toml
78 lines (64 loc) · 1.33 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
73
74
75
76
77
78
[tool.poetry]
name = "entropix"
version = "0.1.0"
description = ""
authors = ['xjdr']
readme = "README.md"
[tool.poetry.dependencies]
python = "^3.11"
tiktoken = "0.4.0"
pydantic = "^2.9.2"
blobfile = "^3.0.0"
ml-dtypes = "^0.5.0"
rich = "^13.8.1"
chex = "^0.1.87"
tyro = "^0.8.11"
jax = {extras = ["cuda12"], version = "^0.4.35"}
[tool.poetry.group.dev.dependencies]
pytest = "^8.3.2"
ruff = "^0.6.2"
transformers = "^4.45.1"
huggingface-hub = {extras = ["cli"], version = "^0.25.1"}
[tool.poetry.group.test.dependencies]
torch = "^2.4.1"
fairscale = "^0.4.13"
[tool.poetry.group.eval.dependencies]
lm-eval = "^0.4.4"
langdetect = "^1.0.9"
immutabledict = "^4.2.0"
anthropic = "^0.39.0"
requests = "^2.32.3"
pandas = "^2.2.3"
blobfile = "^3.0.0"
[tool.poetry.group.server.dependencies]
fastapi = "^0.115.3"
pydantic = "^2.9.2"
uvicorn = "^0.32.0"
openai = "^1.52.2"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.ruff]
extend-exclude = ["tests/**"]
lint.select = [
# pycodestyle
"E",
# Pyflakes
"F",
# pyupgrade
"UP",
# flake8-bugbear
"B",
# flake8-simplify
"SIM",
# isort
"I",
# Ruff
"RUF",
]
lint.ignore = [
"E501", # ignore line-length (e.g. long docstring)
"SIM117", # ignore combine with statements (it leads to harder to read code)
]
line-length = 88
indent-width = 2