effectful/pyproject.toml at master · BasisResearch/effectful · 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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
[build-system]
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "effectful"
version = "0.4.0"
description = "Metaprogramming infrastructure"
readme = "README.rst"
license = "Apache-2.0"
requires-python = ">=3.12"
authors = [
    { name = "Basis" },
]
keywords = [
    "machine learning",
    "statistics",
    "probabilistic programming",
    "bayesian modeling",
    "pytorch",
]
classifiers = [
    "Intended Audience :: Developers",
    "Intended Audience :: Education",
    "Intended Audience :: Science/Research",
    "Operating System :: POSIX :: Linux",
    "Operating System :: MacOS :: MacOS X",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3.13",
    "Programming Language :: Python :: 3.14",
]
dependencies = []

[project.urls]
Homepage = "https://www.basis.ai/"
Source = "https://github.com/BasisResearch/effectful"
"Bug Tracker" = "https://github.com/BasisResearch/effectful/issues"

[project.optional-dependencies]
torch = ["torch>=2.10"]
pyro = ["pyro-ppl>=1.9.1"]
jax = ["jax"]
numpyro = [
  "numpyro>=0.19",
  "jax<0.10"
]
llm = [
  # 1.92.0 ships no wheel, so it builds from sdist; on Python 3.14 that build pulls
  # pyo3-ffi 0.23.5, which doesn't support 3.14, so `uv sync` fails the whole matrix.
  "litellm!=1.92.0",
  "langfuse",
  "rich",
  "tenacity",
  "mypy",
  "ty",
  "pillow",
  "pydantic",
  "annotated-types",
  "typing_extensions",
  "restrictedpython>=8.1",
  "matplotlib>=3.11.1",
  "agent-client-protocol",
  "fastmcp",
]
prettyprinter = ["prettyprinter"]
docs = [
    "effectful[torch,pyro,jax,numpyro,llm,prettyprinter]",
    # For the ACP example under docs/source/llm_examples/acp, which serves an Agent
    # to an editor over the Agent Client Protocol. Imported as `acp`.
    "agent-client-protocol>=0.12,<0.13",
    "sphinx",
    "sphinxcontrib-bibtex",
    "sphinx_rtd_theme",
    "myst-parser",
    "nbsphinx",
    "sphinx_autodoc_typehints>=3.6,<3.9",
    "pypandoc_binary<1.16",
]
test = [
    "effectful[torch,pyro,jax,numpyro,docs,prettyprinter]",
    "pytest",
    "pytest-cov",
    "pytest-xdist",
    "pytest-benchmark",
    "pytest-timeout",
    "mypy",
    "ruff",
    "nbval",
    "nbqa",
]

[dependency-groups]
dev = ["effectful[torch,pyro,jax,numpyro,llm,docs,test]"]

[tool.ruff]
target-version = "py312"

[tool.ruff.lint]
select = ["F", "I", "PERF", "UP"]

[tool.pytest.ini_options]
addopts = "--doctest-modules --dist loadgroup"
testpaths = ["./effectful", "./tests"]
markers = [
    "example: runs a docs example end to end against a live model",
    "nightly_example: a docs example too slow or costly to run on every pull request",
]

[tool.mypy]
ignore_missing_imports = true
warn_unused_ignores = true
allow_empty_bodies = true
exclude = "build|test_internals_unification_typeddict"