jinns/pyproject.toml at main · mia-jinns/jinns · 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
[build-system]
requires = ["setuptools", "setuptools-scm"]
build-backend = "setuptools.build_meta"

[tool.setuptools]
packages = ["jinns"]

[tool.setuptools_scm]
[project]
name = "jinns"
dynamic = ["version"]
description = "Physics Informed Neural Network with JAX"
readme = "README.md"
license = {text = "Apache License 2.0"}
requires-python = ">=3.11"
keywords = []
authors = [
  {name = "Hugo Gangloff", email = "hugo.gangloff@inrae.fr"},
  {name = "Nicolas Jouvin", email = "nicolas.jouvin@inrae.fr"},
]
maintainers = [
  {name = "Hugo Gangloff", email = "hugo.gangloff@inrae.fr"},
  {name = "Nicolas Jouvin", email = "nicolas.jouvin@inrae.fr"},
]
classifiers = [
  "License :: OSI Approved :: Apache Software License",
  "Development Status :: 5 - Production/Stable",
  "Programming Language :: Python",
]
dependencies = [
    "numpy>=2.0.0",
    "jax>=0.8.1",
    "optax>=0.2.6",
    "equinox>=0.13.2",
    "matplotlib",
    "jaxtyping",
    "pyright>=1.1.407",
    "pytest>=9.0.2",
]

[project.optional-dependencies]
notebook = ["jupyter", "seaborn", "pandas", "diffrax"]
gpu_cuda13 = ["jax[cuda13]>=0.8.1"]
gpu_cuda12 = ["jax[cuda12]>=0.8.1"]
cpu = ["jax>=0.8.1"]
dev = [
  "pre-commit", "pylint"
]
docs = [
  "hippogriffe==0.2.2",
  "griffe==1.7.3",
  "mkdocs==1.6.1",
  "mkdocs-include-exclude-files==0.1.0",
  "mkdocs-ipynb==0.1.1",
  "mkdocs-material==9.6.7",
  "mkdocstrings==0.28.3",
  "mkdocstrings-python==1.16.8",
  "pymdown-extensions==10.14.3",
  "mknotebooks"
]

[project.urls]
Repository = "https://gitlab.com/mia_jinns/jinns"
Documentation = "https://mia_jinns.gitlab.io/jinns/index.html"

[tool.coverage.report]
exclude_also = [
    "def __repr__",
    "raise AssertionError",
    "raise NotImplementedError",
    "if __name__ == .__main__.:",
    "if TYPE_CHECKING:",
    "@(abc\\.)?abstractmethod",
    # A pragma comment that excludes an entire file:
    "\\A(?s:.*# pragma: exclude file.*)\\Z",
    ]

[tool.pyright]
include = ["jinns"]
exclude = ["jinns/experimental"]
pythonVersion = "3.11"
typeCheckingMode = "basic"

[tool.ruff]
[tool.ruff.lint]
ignore = ["F722", "E741", "E731", "E722"]