-
Notifications
You must be signed in to change notification settings - Fork 33
/
pyproject.toml
115 lines (101 loc) · 2.75 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
[build-system]
requires = ["setuptools", "setuptools-scm"]
build-backend = "setuptools.build_meta"
[project]
name = "makani"
authors = [
{ name="Thorsten Kurth", email = "tkurth@nvidia.com" },
{ name="Boris Bonev", email = "bbonev@nvidia.com" },
{ name="Jean Kossaifi", email = "jkossaifi@nvidia.com" },
{ name="Animashree Anandkumar" },
{ name="Kamyar Azizzadenesheli" },
{ name="Noah Brenowitz" },
{ name="Ashesh Chattopadhyay" },
{ name="Yair Cohen" },
{ name="David Hall" },
{ name="Peter Harrington" },
{ name="Pedram Hassanzadeh" },
{ name="Christian Hundt" },
{ name="Alexey Kamenev"},
{ name="Karthik Kashinath" },
{ name="Zongyi Li" },
{ name="Morteza Mardani" },
{ name="Jaideep Pathak" },
{ name="Mike Pritchard" },
{ name="David Pruitt" },
{ name="Sanjeev Raja" },
{ name="Shashank Subramanian" },
]
maintainers = [
{ name="Thorsten Kurth", email = "tkurth@nvidia.com" },
{ name="Boris Bonev", email = "bbonev@nvidia.com" },
]
description = "Massively parallel training of machine-learning based weather and climate models"
readme = "README.md"
requires-python = ">=3.10"
classifiers = [
"Development Status :: Alpha",
"Programming Language :: Python :: 3.10",
"License :: OSI Approved :: Apache Software License",
"Operating System :: OS Independent",
]
dynamic = ["version"]
dependencies = [
"torch>=2.0.0",
"numpy>=1.22.4,<1.25",
"numba>=0.50.0",
"nvidia_dali_cuda110>=1.16.0",
"nvidia-modulus>=0.5.0a0",
"torch-harmonics>=0.6.5",
"tensorly>=0.8.1",
"tensorly-torch>=0.4.0",
"numba>=0.50.0",
]
[tool.setuptools.dynamic]
version = {attr = "makani.__version__"}
[tool.setuptools.packages.find]
include = ["makani*"]
[project.optional-dependencies]
dev = [
"pytest>=6.0.0",
"black>=22.10.0",
"coverage>=6.5.0",
]
all = [
"wandb>=0.13.7",
"imageio>=2.28.1",
"moviepy>=1.0.3",
"tqdm>=4.60.0",
]
[tool.pytest.ini_options]
minversion = "6.0"
addopts = ""
testpaths = [
"tests",
"makani",
]
[tool.black]
line-length = 120
include = '\.pyi?$'
exclude = '''
/(
\.git
| \.hg
| \.mypy_cache
| \.tox
| \.venv
| _build
| buck-out
| build
| dist
)/
'''
[project.entry-points."makani.models"]
AFNOv1 = "makani.models.networks.afnonet:AdaptiveFourierNeuralOperatorNet"
AFNO = "makani.models.networks.afnonet_v2:AdaptiveFourierNeuralOperatorNet"
FNO = "makani.models.networks.sfnonet:FourierNeuralOperatorNet"
SFNO = "makani.models.networks.sfnonet:SphericalFourierNeuralOperatorNet"
DebugNet = "makani.models.networks.debug:DebugNet"
ViT = "makani.models.networks.vit:VisionTransformer"
[project.entry-points."modulus.models"]
SFNO = "makani.models.networks.sfnonet:SFNO"