rydopt/pyproject.toml at main · dflocher/rydopt · 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
[project]
name = "RydOpt"
version = "0.3.0"
description = "Multiqubit Rydberg Gate Optimizer"
readme = "README.md"
authors = [ {name = "David F. Locher"}, {name = "Sebastian Weber"}, {name = "Jakob Holschbach"}]
maintainers = [ {name = "Sebastian Weber", email = "seweber@posteo.de"}, {name = "David F. Locher", email = "david.locher@rwth-aachen.de"} ]
license = { file = "LICENSE" }
classifiers = [
  "Intended Audience :: Science/Research",
  "License :: OSI Approved :: MIT License",
  "Operating System :: MacOS",
  "Operating System :: Microsoft :: Windows",
  "Operating System :: POSIX :: Linux",
  "Programming Language :: Python",
  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3 :: Only",
  "Programming Language :: Python :: 3.10",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: 3.13",
  "Programming Language :: Python :: Implementation :: CPython",
  "Topic :: Scientific/Engineering",
  "Topic :: Scientific/Engineering :: Physics",
  "Topic :: Scientific/Engineering :: Quantum Computing",
]
requires-python = ">=3.10"
dependencies = [
    "diffrax>=0.5.0",
    "jax>=0.4.28",
    "jaxlib>=0.4.28",
    "matplotlib>=3.9.0",
    "numpy>=2.0.2",
    "scipy>=1.14.1",
    "optax>=0.2.2",
    "qutip>=5.1.0",
    "tqdm>=4.65.0",
    "typing-extensions>=4.15.0",
]

[dependency-groups]
dev = [
    "pytest>=8.4.2",
    "sphinx > 7",
    "sphinx-rtd-theme>=3.0.2",
    "sphinx-autobuild>=2024.10.3",
    "sphinxcontrib-napoleon>=0.7",
    "nbsphinx>=0.9.7",
    "sphinx-tabs @ git+https://github.com/executablebooks/sphinx-tabs.git@143083f922aab49e26638611dce4754d43b29310", # FIXME: install from pypi once https://github.com/executablebooks/sphinx-tabs/pull/207 is included
    "sphinx-copybutton>=0.5.2",
    "notebook>=7.5.0",
    "ipykernel>=7.1.0",
    "ipywidgets>=8.1.8",
    "nbmake>=1.5.5",
]

[project.optional-dependencies]
cuda12 = [
    "jax[cuda12]>=0.4.28",
]

[build-system]
requires = ["uv_build>=0.9.13,<0.10.0"]
build-backend = "uv_build"

[tool.ruff]
line-length = 120
target-version = "py310"
extend-include = ["*.ipynb"]
extend-exclude = ["docs/conf.py"]

[tool.ruff.lint]
select = ["E", "F", "W", "I", "UP", "N", "B", "A", "C4", "SIM", "PT", "RET", "RUF", "TID", "ERA", "D", "ANN"]
ignore = [
    "E741",
    "B905",
    "N803",
    "N802",
    "N806",
    "RUF003",
    "D415",
    "D203",
    "D213",
    "RUF022",
    "D205", # TODO every docstring should start with a one-line description
    "D400", # TODO first line of docstrings should end with a period
    "D401", # TODO first line of docstring should be in imperative mood
    "D102", # TODO add docstrings in all public methods
    "D103", # TODO add docstrings in all public functions
    "D100", # TODO add docstrings in all public modules
    "D104", # TODO add docstrings in public package
    "D101", # TODO add docstrings in public class
    "D107", # TODO add docstrings in __init__
]

[tool.ruff.lint.isort]
combine-as-imports = true
force-wrap-aliases = true

[tool.ruff.lint.per-file-ignores]
"docs/examples/**" = ["I001"]

[tool.ruff.format]
docstring-code-format = true

[tool.pytest.ini_options]
filterwarnings = [
  # JAX deprecation from optax # FIXME: remove once optax is updated
  "ignore:Setting `jax_pmap_shmap_merge` is deprecated:DeprecationWarning:^jax\\._src\\.config$",
  # Diffrax complex dtype warning coming via equinox._jit # FIXME: remove once equinox is updated
  "ignore:Complex dtype support in Diffrax.*:UserWarning:^equinox\\._jit$",
]