Skip to content

Commit

Permalink
py-keras: add v3.8.0, remove optional deps (spack#48461)
Browse files Browse the repository at this point in the history
  • Loading branch information
adamjstewart authored Jan 9, 2025
1 parent f4d2ff0 commit 72783bc
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions var/spack/repos/builtin/packages/py-keras/package.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ class PyKeras(PythonPackage):
maintainers("adamjstewart")
license("Apache-2.0")

version("3.8.0", sha256="6289006e6f6cb2b68a563b58cf8ae5a45569449c5a791df6b2f54c1877f3f344")
version("3.7.0", sha256="a4451a5591e75dfb414d0b84a3fd2fb9c0240cc87ebe7e397f547ce10b0e67b7")
version("3.6.0", sha256="405727525a3522ed8f9ec0b46e0667e4c65fcf714a067322c16a00d902ded41d")
version("3.5.0", sha256="53ae4f9472ec9d9c6941c82a3fda86969724ace3b7630a94ba0a1f17ba1065c3")
Expand Down Expand Up @@ -64,6 +65,7 @@ class PyKeras(PythonPackage):
version("2.2.1", sha256="0d3cb14260a3fa2f4a5c4c9efa72226ffac3b4c50135ba6edaf2b3d1d23b11ee")
version("2.2.0", sha256="5b8499d157af217f1a5ee33589e774127ebc3e266c833c22cb5afbb0ed1734bf")

# TODO: add openvino backend (keras 3.8+)
variant(
"backend",
default="tensorflow",
Expand All @@ -85,30 +87,28 @@ class PyKeras(PythonPackage):
depends_on("py-absl-py", when="@2.6:")
depends_on("py-numpy")
depends_on("py-rich", when="@3:")
depends_on("[email protected]:", when="@3.3.3:")
depends_on("py-namex", when="@3:")
depends_on("py-h5py")
depends_on("py-optree", when="@3.1:")
depends_on("py-ml-dtypes", when="@3.0.5:")
depends_on("py-packaging", when="@3.4:")

# requirements-common.txt
depends_on("py-scipy")
depends_on("py-pandas")
depends_on("py-requests", when="@3:")
depends_on("py-protobuf", when="@3:")
# Many more (optional?) dependencies

# requirements-tensorflow-cuda.txt
with when("backend=tensorflow"):
depends_on("[email protected]", when="@3.7:")
depends_on("[email protected]", when="@3.5:3.6")
depends_on("[email protected]:2.16", when="@3.0:3.4")
# depends_on("py-tf2onnx", when="@3.8:")

# requirements-jax-cuda.txt
with when("backend=jax"):
depends_on("[email protected]", when="@3.6:")
depends_on("[email protected]", when="@3.0.5:3.5")
depends_on("py-jax", when="@3:")
# depends_on("py-flax", when="@3.2:")

# requirements-torch-cuda.txt
with when("backend=torch"):
Expand All @@ -126,6 +126,7 @@ class PyKeras(PythonPackage):
depends_on("[email protected]", when="@3.0.3:3.0.5")
depends_on("[email protected]", when="@3.0.1:3.0.2")
depends_on("[email protected]", when="@3.0.0")
# depends_on("py-torch-xla", when="@3.8:")

# Historical dependencies
with default_args(type="build"):
Expand Down

0 comments on commit 72783bc

Please sign in to comment.