summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAdam J. Stewart <ajstewart426@gmail.com>2024-07-15 22:12:16 +0200
committerGitHub <noreply@github.com>2024-07-15 22:12:16 +0200
commit35bd21fc641763ee4f75fc5a13819c14927da083 (patch)
treec1acd8a82facd9f7ca66b1428d3b8f85c525bbd6
parent652170fb54bdd58b22f7487bf2678e3e053c1e77 (diff)
downloadspack-35bd21fc641763ee4f75fc5a13819c14927da083.tar.gz
spack-35bd21fc641763ee4f75fc5a13819c14927da083.tar.bz2
spack-35bd21fc641763ee4f75fc5a13819c14927da083.tar.xz
spack-35bd21fc641763ee4f75fc5a13819c14927da083.zip
py-tensorflow-estimator: correct dependencies (#44185)
-rw-r--r--var/spack/repos/builtin/packages/py-tensorflow-estimator/package.py61
1 files changed, 32 insertions, 29 deletions
diff --git a/var/spack/repos/builtin/packages/py-tensorflow-estimator/package.py b/var/spack/repos/builtin/packages/py-tensorflow-estimator/package.py
index eba0f1dbe9..02ed475169 100644
--- a/var/spack/repos/builtin/packages/py-tensorflow-estimator/package.py
+++ b/var/spack/repos/builtin/packages/py-tensorflow-estimator/package.py
@@ -9,8 +9,9 @@ from spack.package import *
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"))
-
- 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")
+ with default_args(type="build"):
+ depends_on("bazel@0.19.0:")
+ depends_on("py-pip")
+ depends_on("py-wheel")
+
+ # 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")