py-tensorflow-estimator: correct dependencies (#44185)
This commit is contained in:
parent
652170fb54
commit
35bd21fc64
@ -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")
|
||||
|
Loading…
Reference in New Issue
Block a user