goat/pyproject.toml at main · deepvk/goat · 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
58
59
60
61
62
63
64
65
[metadata]
name = "goat"
author = "deepvk, VyrodovMikhail"
url = "https://github.com/deepvk/goat"

[project]
name = "goat"
version="0.0.0"
authors = [
    {name="deepvk, VyrodovMikhail"}
]
description = "A framework for evaluating language models"
readme = "README.md"
license = { "text" = "Apache-2.0" }
requires-python = ">=3.8"
dependencies = [
    "typer<0.10.0",
]

[project.optional-dependencies]
frontend = ["gradio", "psycopg2>=2.9.9", "SQLAlchemy>=2.0.29",]
backend = [
    "psycopg2>=2.9.9",
    "SQLAlchemy>=2.0.29",
    "torch==2.2.0",
    "torchdata==0.7.1",
    "torchtext==0.17.0",
    "torchvision==0.17.0",
    "transformer_engine==0.0.0",
    "lm_eval@git+https://github.com/deepvk/lm-evaluation-harness@goat",
    "fschat[model_worker,llm_judge]@git+https://github.com/deepvk/FastChat",
]
parser = [
    "packaging",
    "numpy",
    "pandas",
    "requests",
    "Scrapy>=2.11.0",
]
all = ["goat[frontend]", "goat[backend]", "goat[parser]"]

[tool.isort]
profile = "black"
line_length = 120

[tool.black]
line-length = 120

[tool.mypy]
files = ["goat"]
install_types = "True"
non_interactive = "True"
disallow_untyped_defs = "True"
ignore_missing_imports = "True"
show_error_codes = "True"
warn_redundant_casts = "True"
warn_unused_configs = "True"
warn_unused_ignores = "True"
allow_redefinition = "True"
warn_no_return = "False"
no_implicit_optional = "False"

[tool.pytest.ini_options]
testpaths = ["tests"]
addopts = ["--color=yes", "-s"]