jaxfun/pyproject.toml at main · spectralDNS/jaxfun · 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
[build-system]
build-backend = "uv_build"

requires = [
  "uv-build>=0.8,<0.9",
]

[project]
name = "jaxfun"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
authors = [ { name = "Mikael Mortensen" }, { name = "August Femtehjell" } ]
requires-python = ">=3.12"
classifiers = [
  "Programming Language :: Python :: 3 :: Only",
  "Programming Language :: Python :: 3.12",
]
dependencies = [
  "flax>=0.12",
  "jax>=0.8.1",
  "marimo>=0.19.10",
  "matplotlib>=3.9.3",
  "optax>=0.2.6",
  "pre-commit>=4.3",
  "pytest-xdist[psutil]>=3.8",
  "sympy>=1.14",
]

[tool.pytest.ini_options]
markers = [ "slow: marks tests as slow (select with '-m slow')" ]
addopts = [ "-m", "not slow", "-n", "logical" ]

[tool.coverage.run]
branch = true
omit = [ "examples/*", "tests/*" ]

[tool.coverage.report]
# Regexes for lines to exclude from consideration
exclude_also = [
  # Don't complain about missing debug-only code:
  "def __repr__",
  "if self\\.debug",

  # Don't complain if tests don't hit defensive assertion code:
  "raise AssertionError",
  "raise NotImplementedError",

  # Don't complain if non-runnable code isn't run:
  "if 0:",
  "if __name__ == .__main__.:",

  # Don't complain about abstract methods, they aren't run:
  "@(abc\\.)?abstractmethod",
]

[tool.ty.src]
exclude = [ "sandbox", "examples/notebooks", "**/*.ipynb" ]

[[tool.ty.overrides]]
include = [ "examples/DrivenCavity_lstsq.py" ]
rules = { "invalid-assignment" = "ignore" }

[[tool.ty.overrides]]
include = [ "examples/sphere_helmholtz.py" ]
rules = { "possibly-missing-attribute" = "ignore" }

[dependency-groups]
dev = [
  "ipython>=8.37.0",
  "pytest>=8.4.1",
  "pyvista>=0.46.2",
  "ruff>=0.15.0",
  "soap-jax",
  "mpi4py>=4.1.1",
  "ipykernel>=6.30.1",
  "pytest-cov>=6.2.1",
  "plotly>=6.3.0",
  "marimo[recommended]>=0.19.7",
  "pre-commit>=4.3.0",
  "shapely>=2.1.0",
  "ty>=0.0.24",
  "urllib3>=2.6.3",
]

[tool.uv.sources]
soap-jax = { git = "https://github.com/mikaem/SOAP_JAX" }
# soap-jax = { git = "https://github.com/haydn-jones/SOAP_JAX" }