summaryrefslogtreecommitdiff
path: root/var/spack/repos/builtin/packages/py-tfdlpack/package.py
diff options
context:
space:
mode:
Diffstat (limited to 'var/spack/repos/builtin/packages/py-tfdlpack/package.py')
-rw-r--r--var/spack/repos/builtin/packages/py-tfdlpack/package.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/py-tfdlpack/package.py b/var/spack/repos/builtin/packages/py-tfdlpack/package.py
index ce0771a5f4..e8419866fd 100644
--- a/var/spack/repos/builtin/packages/py-tfdlpack/package.py
+++ b/var/spack/repos/builtin/packages/py-tfdlpack/package.py
@@ -4,7 +4,7 @@
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
-class PyTfdlpack(CMakePackage):
+class PyTfdlpack(CMakePackage, PythonPackage):
"""Tensorflow plugin for DLPack."""
homepage = "https://github.com/VoVAllen/tf-dlpack"
@@ -37,8 +37,8 @@ class PyTfdlpack(CMakePackage):
def install(self, spec, prefix):
with working_dir('python'):
- setup_py('install', '--prefix=' + prefix,
- '--single-version-externally-managed', '--root=/')
+ args = std_pip_args + ['--prefix=' + prefix, '.']
+ pip(*args)
def setup_run_environment(self, env):
# Prevent TensorFlow from taking over the whole GPU