From 1679274ac2c959101ed79495e289e04fbc54a841 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Sat, 29 Jan 2022 20:22:19 -0600 Subject: py-horovod: add py-tensorflow-estimator dependency (#28674) --- var/spack/repos/builtin/packages/py-horovod/package.py | 1 + 1 file changed, 1 insertion(+) diff --git a/var/spack/repos/builtin/packages/py-horovod/package.py b/var/spack/repos/builtin/packages/py-horovod/package.py index 6bd68530e5..eed7ca386d 100644 --- a/var/spack/repos/builtin/packages/py-horovod/package.py +++ b/var/spack/repos/builtin/packages/py-horovod/package.py @@ -63,6 +63,7 @@ class PyHorovod(PythonPackage, CudaPackage): # Framework dependencies depends_on('py-tensorflow@1.1.0:', type=('build', 'link', 'run'), when='frameworks=tensorflow') depends_on('py-tensorflow@1.15:', type=('build', 'link', 'run'), when='frameworks=tensorflow @0.20:') + depends_on('py-tensorflow-estimator', type=('build', 'run'), when='frameworks=tensorflow') depends_on('py-torch@0.4.0:', type=('build', 'link', 'run'), when='frameworks=pytorch') depends_on('py-torch@1.2:', type=('build', 'link', 'run'), when='frameworks=pytorch @0.20:') depends_on('py-torchvision', type=('build', 'run'), when='frameworks=pytorch @:0.19.1') -- cgit v1.2.3-60-g2f50