pyproject.toml

#!/usr/bin/env python
# (C) Copyright 2024 ECMWF.
#
# 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.

# https://packaging.python.org/en/latest/guides/writing-pyproject-toml/

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

[project]
name = "ai-models-gencast"

description = "An ai-models plugin to run Deepmind's gencast model"
readme = "README.md"
keywords = [
  "ai",
  "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.9",
  "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",
]

# JAX requirements are in requirements.txt

dependencies = [
  "ai-models>=0.7.4",
  "dm-tree",
  "dm-haiku",
  "xarray<=2024.11.0" # Locked to avoid the FutureChange in xarray about The return type of `Dataset.dims`
]

optional-dependencies.dev = [
  "pre-commit",
]
urls.Repository = "https://github.com/ecmwf-lab/ai-models-gencast"
entry-points."ai_models.model".gencast = "ai_models_gencast.model:GenCast0p25deg"
entry-points."ai_models.model"."gencast-0.25" = "ai_models_gencast.model:GenCast0p25deg"
entry-points."ai_models.model"."gencast-0.25-Oper" = "ai_models_gencast.model:GenCast0p25degOper"
entry-points."ai_models.model"."gencast-1.0" = "ai_models_gencast.model:GenCast1p0deg"
entry-points."ai_models.model"."gencast-1.0-Mini" = "ai_models_gencast.model:GenCast1p0degMini"

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