py-tensorflow-estimator: correct dependencies (#44185)

This commit is contained in:
Adam J. Stewart 2024-07-15 22:12:16 +02:00 committed by GitHub
parent 652170fb54
commit 35bd21fc64
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -9,8 +9,9 @@
class PyTensorflowEstimator(Package):
"""TensorFlow Estimator is a high-level TensorFlow API that greatly
simplifies machine learning programming."""
"""TensorFlow Estimator is a high-level API that encapsulates
model training, evaluation, prediction, and exporting.
"""
homepage = "https://github.com/tensorflow/estimator"
url = "https://github.com/tensorflow/estimator/archive/v2.2.0.tar.gz"
@ -28,39 +29,41 @@ class PyTensorflowEstimator(Package):
version("2.8.0", sha256="58a2c3562ca6491c257e9a4d9bd8825667883257edcdb452181efa691c586b17")
version("2.7.0", sha256="e5164e802638d3cf110ecc17912be9d514a9d3354ec48e77200b9403dcc15965")
version("2.6.0", sha256="947705c60c50da0b4a8ceec1bc058aaf6bf567a7efdcd50d5173ebf6bafcf30f")
version("2.5.0", sha256="66661f30ea05d57377c45267ca770935fb8c54f85b7901f0a7deb91766fe9f45")
version("2.4.0", sha256="e6ea12014c3d8c89a81ace95f8f8b7c39ffcd3e4e4626709e4aee0010eefd962")
version("2.3.0", sha256="75403e7de7e8ec30ec0781ede56ed84cbe5e90daad64a9c242cd489c8fe63a17")
version("2.2.0", sha256="2d68cb6e6442e7dcbfa2e092aa25bdcb0eda420536a829b85d732854a4c85d46")
extends("python")
# tensorflow_estimator/tools/pip_package/setup.py
depends_on("python@3.7:", when="@2.9:", type=("build", "run"))
with default_args(type="build"):
depends_on("bazel@0.19.0:")
depends_on("py-pip")
depends_on("py-wheel")
for ver in ["2.14", "2.13", "2.12", "2.11", "2.10", "2.9", "2.8", "2.7", "2.6"]:
depends_on("py-keras@" + ver, when="@" + ver, type=("build", "run"))
for ver in [
"2.14",
"2.13",
"2.12",
"2.11",
"2.10",
"2.9",
"2.8",
"2.7",
"2.6",
"2.5",
"2.4",
"2.3",
"2.2",
]:
depends_on("py-tensorflow@" + ver, when="@" + ver, type=("build", "run"))
depends_on("bazel@0.19.0:", type="build")
depends_on("py-pip", type="build")
depends_on("py-wheel", type="build")
# See expect_*_installed in tensorflow_estimator/python/estimator/BUILD
with default_args(type=("build", "run")):
depends_on("py-absl-py")
depends_on("py-h5py")
depends_on("py-numpy")
depends_on("py-pandas")
depends_on("py-six")
for ver in [
"2.14",
"2.13",
"2.12",
"2.11",
"2.10",
"2.9",
"2.8",
"2.7",
"2.6",
"2.4",
"2.3",
"2.2",
]:
depends_on(f"py-tensorboard@{ver}", when=f"@{ver}")
depends_on(f"py-tensorflow@{ver}", when=f"@{ver}")
depends_on(f"py-keras@{ver}", when=f"@{ver}")
def install(self, spec, prefix):
self.tmp_path = tempfile.mkdtemp(prefix="spack")