genlm-eval/pyproject.toml at main · genlm/genlm-eval · 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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
[build-system]
requires = ["setuptools>=64.0", "setuptools-scm>=8"]
build-backend = "setuptools.build_meta"

[tool.setuptools_scm]

[project]
name = "genlm-eval"
dynamic = ["version"]
description = "Evaluation framework for models in the GenLM ecosystem"
authors = []
readme = "README.md"
requires-python = ">=3.11"

dependencies = [
    "pydantic",
    "numpy",
    "pandas",
]

[project.optional-dependencies]
pattern_matching = [
    "genlm-control>=0.2.1",
    "regex",
]

spider = [
    "genlm-control>=0.2.1",
    "jsons",
    "lark",
    "rapidfuzz",
]

molecules = [
    "genlm-control>=0.2.1",
    "rdkit",
    "partialsmiles",
]

ds1000 = [
    "genlm-control>=0.2.1",
    "datasets>=4.0.0",
]

ds1000_code_env = [
  "gensim==4.3.2",
  "matplotlib==3.8.4",
  "numpy==1.26.4",
  "pandas==1.5.3",
  "torch==2.2.0",
  "seaborn==0.13.2",
  "scikit-learn==1.4.0",
  "scipy==1.12.0",
  "statsmodels==0.14.1",
  "xgboost==2.0.3",
  "PyYAML",
  "tensorflow==2.16.1",
]

json_schema_bench = [
    "datasets>=4.0.0",
    "jsonschema"
]

goal_inference = [
    "genlm-control>=0.2.1",
    "datasets>=4.0.0",
    "polars",
    "planetarium @ git+https://github.com/samuki/planetarium.git"
]

test = [
    "genlm-control>=0.2.1",
    "coverage",
    "pytest",
    "pytest-asyncio",
    "pytest-cov",
]

docs = [
    "mkdocs",
    "mkdocs-jupyter",
    "mkdocstrings[python]",
    "mkdocs-material",
    "mkdocs-gen-files",
    "mkdocs-literate-nav",
    "mkdocs-section-index",
]

[tool.setuptools.packages.find]
include = ["genlm", "genlm/eval"]