anemoi-inference/pyproject.toml at main · vegardb/anemoi-inference · 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
113
114
115
116
117
118
119
120
121
122
123
124
# (C) Copyright 2024 Anemoi contributors.
#
# This software is licensed under the terms of the Apache Licence Version 2.0
# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
#
# In applying this licence, ECMWF does not waive the privileges and immunities
# granted to it by virtue of its status as an intergovernmental organisation
# nor does it submit to any jurisdiction.

[build-system]
build-backend = "setuptools.build_meta"

requires = [ "setuptools>=60", "setuptools-scm>=8" ]

[project]
name = "anemoi-inference"

description = "A package to run inference from data-driven forecasts weather models."
readme = "README.md"

keywords = [ "ai", "inference", "tools" ]

license = { file = "LICENSE" }
authors = [
  { name = "European Centre for Medium-Range Weather Forecasts (ECMWF)", email = "software.support@ecmwf.int" },
]

requires-python = ">=3.10"

classifiers = [
  "Development Status :: 4 - Beta",
  "Intended Audience :: Developers",
  "License :: OSI Approved :: Apache Software License",
  "Operating System :: OS Independent",
  "Programming Language :: Python :: 3 :: Only",
  "Programming Language :: Python :: 3.10",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: Implementation :: CPython",
  "Programming Language :: Python :: Implementation :: PyPy",
]

dynamic = [ "version" ]
dependencies = [
  "anemoi-transform>0.1.13",
  "anemoi-utils[provenance,text]>=0.4.32",
  "aniso8601",
  "anytree",
  "deprecation",
  "earthkit-data>=0.12.4",
  "eccodes>=2.38.3",
  "numpy",
  "omegaconf>=2.2,<2.4",
  "packaging",
  "pydantic",
  "pyyaml",
  "rich",
  "semantic-version",
  "torch",
]

optional-dependencies.all = [
  "anemoi-datasets",
  "anemoi-inference[huggingface,plot,tests,zarr]",
  "anemoi-utils[all]>=0.4.32",
]
optional-dependencies.cosmo = [
  "eccodes<=2.39.1",
  "eccodes-cosmo-resources-python",
]

optional-dependencies.dev = [ "anemoi-inference[all,docs,plugin,tests]" ]

optional-dependencies.docs = [
  "autodoc-pydantic",
  "nbsphinx",
  "pandoc",
  "rstfmt",
  "sphinx<8.2",
  "sphinx-argparse<0.5",
  "sphinx-rtd-theme",
]

optional-dependencies.huggingface = [ "huggingface-hub" ]
optional-dependencies.plot = [ "earthkit-plots" ]

optional-dependencies.plugin = [ "ai-models>=0.7", "tqdm" ]
optional-dependencies.tests = [
  "anemoi-datasets[all]",
  "anemoi-inference[all]",
  "hypothesis",
  "pytest",
  "pytest-mock",
]
optional-dependencies.zarr = [ "zarr" ]

urls.Documentation = "https://anemoi-inference.readthedocs.io/"
urls.Homepage = "https://github.com/ecmwf/anemoi-inference/"
urls.Issues = "https://github.com/ecmwf/anemoi-inference/issues"
urls.Repository = "https://github.com/ecmwf/anemoi-inference/"
scripts.anemoi-inference = "anemoi.inference.__main__:main"

entry-points."ai_models.model".anemoi = "anemoi.inference.plugin:AIModelPlugin"

[tool.setuptools.package-data]
"anemoi.inference.grib.templates" = [ "*.yaml" ]

[tool.setuptools.packages.find]
where = [ "src" ]

[tool.setuptools_scm]
version_file = "src/anemoi/inference/_version.py"

[tool.pytest.ini_options]
markers = [
  "skip_on_hpc: mark a test that should not be run on HPC",
]
testpaths = "tests"

[tool.mypy]
exclude = [ "docs/" ]
strict = false
ignore_missing_imports = true
allow_redefinition = true