diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/py-tensorflow-estimator/package.py | 61 |
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") |