zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-lightning-uq-box: add new package (#47132)

Browse Source
This commit is contained in:
Adam J. Stewart 2024-10-24 20:08:18 +02:00 committed by GitHub
parent f09ce00fe1
commit 3804d128e7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
15 changed files with 332 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
var/spack/repos/builtin/packages/py-asdfghjkl/package.py Normal file
View File

@ -0,0 +1,24 @@
# 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 PyAsdfghjkl(PythonPackage):
"""ASDL: Automatic Second-order Differentiation (for Fisher, Gradient covariance, Hessian,
Jacobian, and Kernel) Library."""
homepage = "https://github.com/kazukiosawa/asdl"
pypi = "asdfghjkl/asdfghjkl-0.1a4.tar.gz"
license("MIT")
version("0.1a4", sha256="a934411a0ffdee6fcdccb19672196498ea6a8e55e3e67abbe67200c84b46ddee")
depends_on("py-setuptools@42:")
with default_args(type=("build", "run")):
depends_on("py-torch@1.13:")
depends_on("py-numpy")

27
var/spack/repos/builtin/packages/py-backpack-for-pytorch/package.py Normal file
View File

@ -0,0 +1,27 @@
# 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 PyBackpackForPytorch(PythonPackage):
"""BackPACK: Packing more into backprop."""
homepage = "https://github.com/f-dangel/backpack"
pypi = "backpack-for-pytorch/backpack-for-pytorch-1.6.0.tar.gz"
license("MIT")
version("1.6.0", sha256="af6495b71bacf82a1c7cab01aa85bebabccfe74d87d89f108ea72a4a0d384de3")
with default_args(type="build"):
depends_on("py-setuptools@38.3:")
depends_on("py-setuptools-scm")
with default_args(type=("build", "run")):
depends_on("py-torch@1.9:")
depends_on("py-torchvision@0.7:")
depends_on("py-einops@0.3:0")
depends_on("py-unfoldnd@0.2:0")

29
var/spack/repos/builtin/packages/py-curvlinops-for-pytorch/package.py Normal file
View File

@ -0,0 +1,29 @@
# 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 PyCurvlinopsForPytorch(PythonPackage):
"""scipy Linear operators for curvature matrices in PyTorch."""
homepage = "https://github.com/f-dangel/curvlinops"
pypi = "curvlinops_for_pytorch/curvlinops_for_pytorch-2.0.0.tar.gz"
license("MIT")
version("2.0.0", sha256="01f9925db9454fc9b0a31c7b83fc8ec2534c2eb12b7de7825a5298fc14e460e7")
with default_args(type="build"):
depends_on("py-setuptools@61:")
depends_on("py-setuptools-scm")
with default_args(type=("build", "run")):
depends_on("py-backpack-for-pytorch@1.6:1")
depends_on("py-torch@2:")
depends_on("py-scipy@1.7.1:1")
depends_on("py-tqdm@4.61:4")
depends_on("py-einops")
depends_on("py-einconv")

25
var/spack/repos/builtin/packages/py-einconv/package.py Normal file
View File

@ -0,0 +1,25 @@
# 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 PyEinconv(PythonPackage):
"""Convolutions as tensor contractions (einsums) for PyTorch."""
homepage = "https://github.com/f-dangel/einconv"
pypi = "einconv/einconv-0.1.0.tar.gz"
license("MIT")
version("0.1.0", sha256="6b103881b1268e43d581f285da4fa72b073c95f31b92575133bafed9929b6d98")
with default_args(type="build"):
depends_on("py-setuptools@38.3:")
depends_on("py-setuptools-scm")
with default_args(type=("build", "run")):
depends_on("py-torch")
depends_on("py-einops")

2
var/spack/repos/builtin/packages/py-ema-pytorch/package.py
View File

@ -16,7 +16,9 @@ class PyEmaPytorch(PythonPackage):
license("MIT", checked_by="alex391")
version("0.7.3", sha256="de640f1d1a054c79607aebfcfd4b8dfff1fba1110bf0c8f7d37517637450938a")
version("0.5.1", sha256="e825212a44e8faae5d2cf2a1349961c4416cba0496ffa64d37718d8b06f206b2")
depends_on("py-setuptools", type="build")
depends_on("py-torch@2:", when="@0.7:", type=("build", "run"))
depends_on("py-torch@1.6:", type=("build", "run"))

44
var/spack/repos/builtin/packages/py-gpytorch/package.py
View File

@ -19,6 +19,7 @@ class PyGpytorch(PythonPackage):
license("MIT")
version("1.13", sha256="f4a488633a2a7a4ab37d12553d1d1dd39690043dbceef14ca428b7d5f89f73ba")
version("1.10", sha256="6dc978ab9fbf220a845a4f1ea13104180fc50e6934081f421b37f6120afb7f18")
version("1.9.1", sha256="0bdbba6f6d5957a0f43ef6dc7fec39c47e8a55f632ca33760c6189f259b3ccc3")
version("1.9.0", sha256="a0608184c18a1f518d6a102473427abf00f5351421e12a934530953f6887b34b")
@ -30,21 +31,28 @@ class PyGpytorch(PythonPackage):
version("1.2.0", sha256="fcb216e0c1f128a41c91065766508e91e487d6ffadf212a51677d8014aefca84")
version("1.1.1", sha256="76bd455db2f17af5425f73acfaa6d61b8adb1f07ad4881c0fa22673f84fb571a")
depends_on("python@3.8:", when="@1.9:", type=("build", "run"))
depends_on("python@3.7:", when="@1.7:", type=("build", "run"))
depends_on("python@3.6:", type=("build", "run"))
depends_on("py-setuptools", type="build")
depends_on("py-setuptools-scm", when="@1.9:", type="build")
depends_on("py-torch@1.11:", when="@1.9:", type=("build", "run"))
depends_on("py-torch@1.10:", when="@1.7:", type=("build", "run"))
depends_on("py-torch@1.9:", when="@1.6:", type=("build", "run"))
depends_on("py-torch@1.8.1:", when="@1.5:", type=("build", "run"))
depends_on("py-torch@1.7:", when="@1.3:", type=("build", "run"))
depends_on("py-torch@1.6:", when="@1.2:", type=("build", "run"))
depends_on("py-torch@1.5:", type=("build", "run"))
depends_on("py-scikit-learn", when="@1.2:", type=("build", "run"))
depends_on("py-linear-operator@0.1.1:", when="@1.9:", type=("build", "run"))
depends_on("py-linear-operator@0.2.0:", when="@1.9.1:", type=("build", "run"))
depends_on("py-linear-operator@0.4.0:", when="@1.10:", type=("build", "run"))
depends_on("py-numpy", when="@1.7:1.8", type=("build", "run"))
depends_on("py-scipy", when="@1.2:1.8", type=("build", "run"))
with default_args(type="build"):
depends_on("py-setuptools")
depends_on("py-setuptools-scm", when="@1.9:")
with default_args(type=("build", "run")):
depends_on("py-torch@2:", when="@1.13:")
depends_on("py-torch@1.11:", when="@1.9:")
depends_on("py-torch@1.10:", when="@1.7:")
depends_on("py-torch@1.9:", when="@1.6:")
depends_on("py-torch@1.8.1:", when="@1.5:")
depends_on("py-torch@1.7:", when="@1.3:")
depends_on("py-torch@1.6:", when="@1.2:")
depends_on("py-torch@1.5:")
depends_on("py-jaxtyping@0.2.19", when="@1.13:")
depends_on("py-mpmath@0.19:1.3", when="@1.12:")
depends_on("py-scikit-learn", when="@1.2:")
depends_on("py-scipy@1.6:", when="@1.13:")
depends_on("py-scipy", when="@1.2:1.8")
depends_on("py-linear-operator@0.5.3:", when="@1.13:")
depends_on("py-linear-operator@0.1.1:", when="@1.9:")
depends_on("py-linear-operator@0.2.0:", when="@1.9.1:")
depends_on("py-linear-operator@0.4.0:", when="@1.10:")
# Historical dependencies
depends_on("py-numpy", when="@1.7:1.8")

12
var/spack/repos/builtin/packages/py-h5py/package.py
View File

@ -18,6 +18,8 @@ class PyH5py(PythonPackage):
license("BSD-3-Clause")
version("master", branch="master")
version("3.12.1", sha256="326d70b53d31baa61f00b8aa5f95c2fcb9621a3ee8365d770c551a13dbbcbfdf")
version("3.12.0", sha256="00955a079e9f86c5ae2cd08accb54396c69cda87152312ddd1528e3f90acc866")
version("3.11.0", sha256="7b7e8f78072a2edec87c9836f25f34203fd492a4475709a18b417a33cfb21fa9")
version("3.10.0", sha256="d93adc48ceeb33347eb24a634fb787efc7ae4644e6ea4ba733d099605045c049")
version("3.9.0", sha256="e604db6521c1e367c6bd7fad239c847f53cc46646f2d2651372d05ae5e95f817")
@ -40,11 +42,12 @@ class PyH5py(PythonPackage):
version("2.5.0", sha256="9833df8a679e108b561670b245bcf9f3a827b10ccb3a5fa1341523852cfac2f6")
version("2.4.0", sha256="faaeadf4b8ca14c054b7568842e0d12690de7d5d68af4ecce5d7b8fc104d8e60")
depends_on("c", type="build") # generated
depends_on("c", type="build")
variant("mpi", default=True, description="Build with MPI support")
# Python versions
depends_on("python@3.9:", type=("build", "run"), when="@3.12:")
depends_on("python@:3.9", type=("build", "run"), when="@:2.8")
# Build dependencies
@ -60,9 +63,9 @@ class PyH5py(PythonPackage):
depends_on("py-setuptools", type="build")
# Build and runtime dependencies
depends_on("py-numpy@1.17.3:", type=("build", "run"), when="@3.9:")
depends_on("py-numpy@1.19.3:", type=("build", "run"), when="@3:3.5 ^python@3.9.0:")
depends_on("py-numpy@1.19.3:", type=("build", "run"), when="@3:3.5,3.12: ^python@3.9.0:")
depends_on("py-numpy@1.17.5:", type=("build", "run"), when="@3:3.5 ^python@3.8.0:3.8")
depends_on("py-numpy@1.17.3:", type=("build", "run"), when="@3.9:3.11")
depends_on("py-numpy@1.14.5:", type=("build", "run"), when="@3:")
depends_on("py-numpy@1.7:", type=("build", "run"), when="@:2")
# https://github.com/h5py/h5py/issues/2353
@ -70,7 +73,8 @@ class PyH5py(PythonPackage):
# Link dependencies (py-h5py v2 cannot build against HDF5 1.12 regardless
# of API setting)
depends_on("hdf5@1.10.4:1.14 +hl", when="@3.10:")
depends_on("hdf5@1.10.6:1.14 +hl", when="@3.12:")
depends_on("hdf5@1.10.4:1.14 +hl", when="@3.10:3.11")
depends_on("hdf5@1.8.4:1.14 +hl", when="@3.8:3.9")
depends_on("hdf5@1.8.4:1.12 +hl", when="@3:3.7")
depends_on("hdf5@1.8.4:1.11 +hl", when="@:2")

13
var/spack/repos/builtin/packages/py-jaxtyping/package.py
View File

@ -15,7 +15,16 @@ class PyJaxtyping(PythonPackage):
license("Apache-2.0")
version("0.2.33", sha256="9a9cfccae4fe05114b9fb27a5ea5440be4971a5a075bbd0526f6dd7d2730f83e")
version("0.2.19", sha256="21ff4c3caec6781cadfe980b019dde856c1011e17d11dfe8589298040056325a")
depends_on("python@3.9:", type=("build", "run"))
depends_on("py-hatchling", type="build")
depends_on("py-typeguard@2.13.3", type=("build", "run"))
with default_args(type=("build", "run")):
depends_on("python@3.9:3", when="@0.2.33:")
depends_on("python@3.8:3", when="@0.2.19")
depends_on("py-typeguard@2.13.3", when="@0.2.33:")
depends_on("py-typeguard@2.13.3:", when="@0.2.19")
# Historical dependencies
depends_on("py-numpy@1.12:", when="@0.2.19")
depends_on("py-typing-extensions@3.7.4.1:", when="@0.2.19")

31
var/spack/repos/builtin/packages/py-laplace-torch/package.py Normal file
View File

@ -0,0 +1,31 @@
# 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 PyLaplaceTorch(PythonPackage):
"""laplace - Laplace approximations for deep learning."""
homepage = "https://github.com/aleximmer/Laplace"
pypi = "laplace_torch/laplace_torch-0.2.1.tar.gz"
license("MIT")
version("0.2.1", sha256="641823a6d3e1dcb8297202b896ae2969334bf96df9a4a6f8cf688896d67d96f2")
with default_args(type="build"):
depends_on("py-setuptools@42:")
depends_on("py-setuptools-scm")
with default_args(type=("build", "run")):
depends_on("py-torch@2:")
depends_on("py-torchvision")
depends_on("py-torchaudio")
depends_on("py-backpack-for-pytorch")
depends_on("py-asdfghjkl@0.1a4")
depends_on("py-torchmetrics")
depends_on("py-opt-einsum")
depends_on("py-curvlinops-for-pytorch@2:")

47
var/spack/repos/builtin/packages/py-lightning-uq-box/package.py Normal file
View File

@ -0,0 +1,47 @@
# 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 PyLightningUqBox(PythonPackage):
"""Lighning-UQ-Box: A toolbox for uncertainty quantification in deep learning."""
homepage = "https://github.com/lightning-uq-box/lightning-uq-box"
pypi = "lightning-uq-box/lightning-uq-box-0.1.0.tar.gz"
git = "https://github.com/lightning-uq-box/lightning-uq-box.git"
license("Apache-2.0")
maintainers("nilsleh", "adamjstewart")
version("main", branch="main")
version("0.1.0", sha256="ce44860db75b4fbe487a009bee91c886be2e1835edee93479a6a8633ef2152b1")
depends_on("py-setuptools@61:", type="build")
with default_args(type=("build", "run")):
depends_on("python@3.10:", when="@0.2:")
depends_on("python@3.9:")
depends_on("py-einops@0.3:")
depends_on("py-lightning@2.4:", when="@0.2:")
depends_on("py-lightning@2.1.1:")
depends_on("py-matplotlib@3.5:", when="@0.2:")
depends_on("py-matplotlib@3.3.3:")
depends_on("py-numpy@1.21.1:", when="@0.2:")
depends_on("py-numpy@1.19.3:")
depends_on("py-pandas@1.1.3:")
depends_on("py-torch@2:")
depends_on("py-torchmetrics@1.2:")
depends_on("py-torchvision@0.16.1:")
depends_on("py-scikit-learn@1.3:")
depends_on("py-gpytorch@1.11:")
depends_on("py-laplace-torch@0.2.1:", when="@0.2:")
depends_on("py-laplace-torch@0.1:")
depends_on("py-uncertainty-toolbox@0.1.1:")
depends_on("py-kornia@0.6.9:")
depends_on("py-timm@0.9.2:")
depends_on("py-torchseg@0.0.1:")
depends_on("py-h5py@3.12.1:", when="@0.2:")
depends_on("py-ema-pytorch@0.7:", when="@0.2:")

3
var/spack/repos/builtin/packages/py-linear-operator/package.py
View File

@ -17,6 +17,7 @@ class PyLinearOperator(PythonPackage):
license("MIT")
version("0.5.3", sha256="16122661cd8b8a89ea965c845f650affe0f688f315893bb8dfa1182f148a1a41")
version("0.4.0", sha256="7c57c9f8f258c9785c0db4dd7625f4dd03a340313d7314cba0b633644909f5c6")
version("0.3.0", sha256="84bf572631a7e1576de6920d81600ca0fedcf6bda2f29dbaf440d6e72ce6abab")
version("0.1.1", sha256="81adc1aea9e98f3c4f07f5608eb77b689bc61793e9beebfea82155e9237bf1be")
@ -26,3 +27,5 @@ class PyLinearOperator(PythonPackage):
depends_on("py-setuptools-scm", type="build")
depends_on("py-torch@1.11:", type=("build", "run"))
depends_on("py-scipy", type=("build", "run"))
depends_on("py-jaxtyping@0.2.19", when="@0.5.3:", type=("build", "run"))
depends_on("py-mpmath@0.19:1.3", when="@0.5.3:", type=("build", "run"))

36
var/spack/repos/builtin/packages/py-timm/package.py
View File

@ -12,10 +12,10 @@ class PyTimm(PythonPackage):
homepage = "https://github.com/rwightman/pytorch-image-models"
pypi = "timm/timm-0.4.12.tar.gz"
license("Apache-2.0")
maintainers("adamjstewart")
license("Apache-2.0")
version("1.0.11", sha256="a005f72b87e67ed30cdbf405a9ffd4e723360c780a43b1cefe266af8ecc9d151")
version("0.9.7", sha256="2bfb1029e90b72e65eb9c75556169815f2e82257eaa1f6ebd623a4b4a52867a2")
version("0.9.5", sha256="669835f0030cfb2412c464b7b563bb240d4d41a141226afbbf1b457e4f18cff1")
version("0.9.2", sha256="d0977cc5e02c69bda979fca8b52aa315a5f2cb64ebf8ad2c4631b1e452762c14")
@ -26,20 +26,22 @@ class PyTimm(PythonPackage):
version("0.5.4", sha256="5d7b92e66a76c432009aba90d515ea7a882aae573415a7c5269e3617df901c1f")
version("0.4.12", sha256="b14be70dbd4528b5ca8657cf5bc2672c7918c3d9ebfbffe80f4785b54e884b1e")
# https://github.com/huggingface/pytorch-image-models/commit/f744bda994ec305a823483bf52a20c1440205032
depends_on("python@3.8:", when="@0.9.0", type=("build", "run"))
# https://github.com/huggingface/pytorch-image-models/issues/1530
# https://github.com/huggingface/pytorch-image-models/pull/1649
depends_on("python@:3.10", when="@:0.6.12", type=("build", "run"))
with default_args(type="build"):
depends_on("py-pdm-backend", when="@1:")
depends_on("py-setuptools", when="@:0")
depends_on("py-setuptools", type="build")
depends_on("py-torch@1.7:", when="@0.6:", type=("build", "run"))
depends_on("py-torch@1.4:", type=("build", "run"))
depends_on("py-torchvision", type=("build", "run"))
depends_on("py-pyyaml", when="@0.6:", type=("build", "run"))
depends_on("py-huggingface-hub", when="@0.6:", type=("build", "run"))
depends_on("py-safetensors", when="@0.9:", type=("build", "run"))
with default_args(type=("build", "run")):
# https://github.com/huggingface/pytorch-image-models/issues/1530
# https://github.com/huggingface/pytorch-image-models/pull/1649
depends_on("python@:3.10", when="@:0.6.12", type=("build", "run"))
# https://github.com/rwightman/pytorch-image-models/pull/1256
depends_on("pil@:9", when="@:0.5", type=("build", "run"))
depends_on("py-numpy", when="@:0.5", type=("build", "run"))
depends_on("py-torch@1.7:", when="@0.6:", type=("build", "run"))
depends_on("py-torch@1.4:", type=("build", "run"))
depends_on("py-torchvision", type=("build", "run"))
depends_on("py-pyyaml", when="@0.6:", type=("build", "run"))
depends_on("py-huggingface-hub", when="@0.6:", type=("build", "run"))
depends_on("py-safetensors", when="@0.9:", type=("build", "run"))
# https://github.com/rwightman/pytorch-image-models/pull/1256
depends_on("pil@:9", when="@:0.5", type=("build", "run"))
depends_on("py-numpy", when="@:0.5", type=("build", "run"))

26
var/spack/repos/builtin/packages/py-torchseg/package.py Normal file
View File

@ -0,0 +1,26 @@
# 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 PyTorchseg(PythonPackage):
"""TorchSeg: Semantic Segmentation models for PyTorch."""
homepage = "https://github.com/isaaccorley/torchseg"
pypi = "torchseg/torchseg-0.0.1a4.tar.gz"
maintainers("isaaccorley", "adamjstewart")
license("MIT")
version("0.0.1a4", sha256="4742551753599af92f9f85e5ca6b149b474ffd458bad1aad6b3aad246a3bf4ea")
depends_on("py-setuptools@61:")
with default_args(type=("build", "run")):
depends_on("py-einops@0.7:")
depends_on("py-timm@0.9.12:")
depends_on("py-torch@1.13:")

29
var/spack/repos/builtin/packages/py-uncertainty-toolbox/package.py Normal file
View File

@ -0,0 +1,29 @@
# 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 PyUncertaintyToolbox(PythonPackage):
"""Uncertainty Toolbox: a python toolbox for predictive uncertainty quantification,
calibration, metrics, and visualization."""
homepage = "https://uncertainty-toolbox.github.io/"
pypi = "uncertainty-toolbox/uncertainty-toolbox-0.1.1.tar.gz"
license("MIT")
version("0.1.1", sha256="d9389112bd431edc8b6e44c5b12405dea8f86063ff9b79f0bb178e5ac76bcfa5")
with default_args(type="build"):
depends_on("py-flit-core@3.2:3")
depends_on("py-setuptools")
with default_args(type=("build", "run")):
depends_on("py-numpy@1.19:")
depends_on("py-scipy@1.5:")
depends_on("py-matplotlib@3.2.2:")
depends_on("py-scikit-learn@0.23.1:")
depends_on("py-tqdm@4.54:")

25
var/spack/repos/builtin/packages/py-unfoldnd/package.py Normal file
View File

@ -0,0 +1,25 @@
# 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 PyUnfoldnd(PythonPackage):
"""N-dimensional unfold (im2col) and fold (col2im) in PyTorch."""
homepage = "https://github.com/f-dangel/unfoldNd"
pypi = "unfoldnd/unfoldnd-0.2.2.tar.gz"
license("MIT")
version("0.2.2", sha256="e8fdffeb68bc1b393ddc1b1c87056e0e4616db992e95c7dbc3dc90d564599397")
with default_args(type="build"):
depends_on("py-setuptools@38.3:")
depends_on("py-setuptools-scm")
with default_args(type=("build", "run")):
depends_on("py-torch")
depends_on("py-numpy")