-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathpyproject.toml
115 lines (106 loc) · 2.92 KB
/
pyproject.toml
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
114
115
[tool.poetry]
name = "lagrangebench"
version = "0.2.0"
description = "LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite"
authors = [
"Artur Toshev, Gianluca Galletti <[email protected]>"
]
license = "MIT"
readme = "README.md"
homepage = "https://lagrangebench.readthedocs.io/"
documentation = "https://lagrangebench.readthedocs.io/"
repository = "https://github.com/tumaer/lagrangebench"
keywords = [
"smoothed-particle-hydrodynamics",
"benchmark-suite",
"lagrangian-dynamics",
"graph-neural-networks",
"lagrangian-particles",
]
classifiers = [
"Development Status :: 3 - Alpha",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Operating System :: MacOS",
"Operating System :: POSIX :: Linux",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Scientific/Engineering :: Physics",
"Topic :: Scientific/Engineering :: Hydrology",
"Topic :: Software Development :: Libraries :: Python Modules",
"Typing :: Typed",
]
[tool.poetry.dependencies]
python = ">=3.9,<=3.11"
cloudpickle = ">=2.2.1"
h5py = ">=3.9.0"
PyYAML = ">=6.0"
numpy = ">=1.24.4"
wandb = ">=0.15.11"
pyvista = ">=0.42.2"
jax = {version = "0.4.29", extras = ["cpu"]}
jaxlib = "0.4.29"
dm-haiku = ">=0.0.10"
e3nn-jax = "0.20.3"
jmp = ">=0.0.4"
jraph = "0.0.6.dev0"
optax = "0.1.7"
ott-jax = ">=0.4.2"
matscipy = ">=0.8.0"
torch = {version = "2.3.1+cpu", source = "torchcpu"}
wget = ">=3.2"
omegaconf = ">=2.3.0"
jax-sph = ">=0.0.3"
[tool.poetry.group.dev.dependencies]
# mypy = ">=1.8.0" - consider in the future
pre-commit = ">=3.3.1"
pytest = ">=7.3.1"
pytest-cov = ">=4.1.0"
ruff = "0.2.2"
ipykernel = ">=6.25.1"
[tool.poetry.group.docs.dependencies]
sphinx = "7.2.6"
sphinx-rtd-theme = "1.3.0"
toml = ">=0.10.2"
[[tool.poetry.source]]
name = "torchcpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "explicit"
[tool.ruff]
exclude = [
".git",
".venv",
"venv",
"docs/_build",
"dist",
"notebooks",
]
show-fixes = true
line-length = 88
[tool.ruff.lint]
ignore = ["F811", "E402"]
select = [
"E", # pycodestyle
"F", # Pyflakes
"SIM", # flake8-simplify
"I", # isort
# "D", # pydocstyle - consider in the future
]
[tool.ruff.lint.isort]
known-third-party = ["wandb"]
[tool.pytest.ini_options]
testpaths = "tests/"
addopts = "--cov=lagrangebench --cov-fail-under=50"
filterwarnings = [
# ignore all deprecation warnings except from lagrangebench
"ignore::DeprecationWarning:^(?!.*lagrangebench).*"
]
# Install bumpversion with: pip install -U poetry-bumpversion
# Use: poetry version {major|minor|patch}
[tool.poetry_bumpversion.file."lagrangebench/__init__.py"]
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"