MySRAgent/pyproject.toml at master · tsinghua-fib-lab/MySRAgent · 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
[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "sr-agent"
version = "1.0.1"
description = "Symbolic Regression Agent powered by LLM"
readme = "README.md"
requires-python = ">=3.12"
license = {text = "MIT"}
authors = [
    {name = "YuMeow", email = "yuzh19@tsinghua.org.cn"}
]
dependencies = [
    "numpy>=1.24.0",
    "scipy>=1.10.0",
    "pandas>=2.0.0",
    "joblib>=1.3.0",
    "docstring-parser>=0.16",
    "pyyaml>=6.0",
    "rich>=13.0.0",
    "matplotlib>=3.7.0",
    "openai>=1.0.0",
    "google-genai>=1.0.0",
    "requests>=2.28.0",
    "python-dotenv>=1.0.0",
    "h5py>=3.16.0",
    "datasets>=4.8.0",
]

[project.optional-dependencies]
nn = [
    "torch>=2.0.0",
    "torch-geometric>=2.4.0",
]
dev = [
    "pytest>=8.0.0",
    "pytest-cov>=4.0.0",
]
all = [
    "sr-agent[nn,dev]",
]

[tool.setuptools.packages.find]
where = ["src"]
include = ["sr_agent*"]

[tool.setuptools.package-data]
sr_agent = [
    "*.md",
    "*.yaml",
    "**/*.md",
    "**/*.yaml",
]

[tool.pytest.ini_options]
minversion = "7.0"
testpaths = ["tests"]
python_files = "test_*.py"
python_classes = "Test*"
python_functions = "test_*"
markers = [
    "slow: marks tests as slow (integration tests)",
    "paid: marks tests as paid API calls (requires credits)",
]
addopts = "--quiet -m 'not slow and not paid'"

[tool.ruff]
line-length = 100
target-version = "py312"

[tool.ruff.lint]
select = ["E", "F", "W"]