deeponet-acoustic-wave-prop/pyproject.toml at main · dtu-act/deeponet-acoustic-wave-prop · 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
[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "deeponet_acoustics"
version = "1.1.0"
description = "DeepONet for acoustic wave propagation"
authors = [
    { name = "Nikolas Borrel-Jensen", email = "nikolasborrel@proton.me" },
]
readme = "README.md"
requires-python = ">=3.11"
dependencies = [
    "jax[cpu]>=0.6.2",
    "jaxlib",
    "flax>=0.6.10",
    "optax>=0.2.6",
    "orbax-checkpoint",
    "tensorboard",
    "tensorboard_plugin_profile",
    "torch>=2.8.0",
    "pytorch_lightning",
    "matplotlib",
    "smt",
    "pydot",
    "graphviz",
    "h5py",
    "tqdm",
    "meshio",
    "h5py",
    "ruff",
    "pre-commit",
]

[project.optional-dependencies]
cuda12 = [
    "jax[local_cuda12]==0.7.2",
    "jaxlib==0.7.2",
]
test = [
    "pytest>=7.0.0",
    "pytest-cov>=4.0.0",
]

[tool.ruff]
   line-length = 88
   lint.select = ["E", "F", "I"]  # Error, pyflakes, isort rules
   lint.ignore = ["E501", "E731"]  # Ignore: line too long, lambda assignments

[tool.setuptools.packages.find]
exclude = ["deeponet_acoustics.scripts*"]

[project.scripts]
deeponet-train = "deeponet_acoustics.main_train:main"
deeponet-infer = "deeponet_acoustics.main_inference:main"