py-climax: add new package (#43121)

var/spack/repos/builtin/packages/py-cf-xarray/package.py

@@ -0,0 +1,22 @@
+# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+from spack.package import *
+
+
+class PyCfXarray(PythonPackage):
+    """A convenience wrapper for using CF attributes on xarray objects."""
+
+    homepage = "https://cf-xarray.readthedocs.io/"
+    pypi = "cf_xarray/cf_xarray-0.9.0.tar.gz"
+
+    license("Apache-2.0")
+
+    version("0.9.0", sha256="01213bdc5ed4d41eeb5da179d99076f49a905b1995daef2a0c7ec402b148675c")
+
+    depends_on("python@3.9:", type=("build", "run"))
+    depends_on("py-setuptools@45:", type="build")
+    depends_on("py-setuptools-scm@6.2:+toml", type="build")
+    depends_on("py-xarray@2022.03:", type=("build", "run"))

var/spack/repos/builtin/packages/py-climax/package.py

@@ -0,0 +1,36 @@
+# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+from spack.package import *
+
+
+class PyClimax(PythonPackage):
+    """ClimaX: A foundation model for weather and climate."""
+
+    homepage = "https://github.com/microsoft/ClimaX"
+    url = "https://github.com/microsoft/ClimaX/archive/refs/tags/v0.3.1.tar.gz"
+    git = "https://github.com/microsoft/ClimaX.git"
+
+    license("MIT")
+
+    version("main", branch="main")
+    version("0.3.1", sha256="1a8ab02fd1083de4340e26889ceea75f9dbc6e56433c731ba616cb46767872fc")
+
+    # pyproject.toml
+    depends_on("py-setuptools", type="build")
+
+    # docker/environment.yml
+    # (only including deps that are actually imported, ignoring version)
+    with default_args(type=("build", "run")):
+        depends_on("py-click")
+        depends_on("py-numpy")
+        depends_on("py-pytorch-lightning")
+        depends_on("py-scipy")
+        depends_on("py-timm")
+        depends_on("py-torch")
+        depends_on("py-torchdata")
+        depends_on("py-tqdm")
+        depends_on("py-xarray")
+        depends_on("py-xesmf")

var/spack/repos/builtin/packages/py-xesmf/package.py

@@ -0,0 +1,32 @@
+# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+from spack.package import *
+
+
+class PyXesmf(PythonPackage):
+    """Universal Regridder for Geospatial Data."""
+
+    homepage = "https://github.com/pangeo-data/xESMF"
+    pypi = "xesmf/xesmf-0.8.4.tar.gz"
+
+    license("MIT")
+
+    version("0.8.4", sha256="c5a2c4b3e8dbbc9fccd5772a940f9067d68e824215ef87ba222b06718c4eeb56")
+
+    with default_args(type="build"):
+        depends_on("py-setuptools@41.2:")
+        depends_on("py-setuptools-scm")
+
+    with default_args(type=("build", "run")):
+        depends_on("py-cf-xarray@0.5.1:")
+        # TODO: add optional dependency
+        # https://github.com/esmf-org/esmf/tree/develop/src/addon/esmpy
+        # depends_on("py-esmpy@8:")
+        depends_on("py-numba@0.55.2:")
+        depends_on("py-numpy@1.16:")
+        depends_on("py-shapely")
+        depends_on("py-sparse@0.8:")
+        depends_on("py-xarray@0.16.2:")