From e575a38d967cd20c44fa4eb515d543f0a21127ec Mon Sep 17 00:00:00 2001 From: Toyohisa Kameyama Date: Wed, 2 Dec 2020 23:45:30 +0900 Subject: py-tensorlow: use mkdtemp for temporary directory. (#20212) --- var/spack/repos/builtin/packages/py-tensorflow/package.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/py-tensorflow/package.py b/var/spack/repos/builtin/packages/py-tensorflow/package.py index a5c285fa01..d0b21d38e5 100644 --- a/var/spack/repos/builtin/packages/py-tensorflow/package.py +++ b/var/spack/repos/builtin/packages/py-tensorflow/package.py @@ -4,6 +4,7 @@ # SPDX-License-Identifier: (Apache-2.0 OR MIT) import sys +import tempfile class PyTensorflow(Package, CudaPackage): @@ -523,8 +524,7 @@ class PyTensorflow(Package, CudaPackage): # ]) # to not be nfs. This is only valid for Linux and we'd like to # stay at least also OSX compatible - tmp_path = '/tmp/spack/tf' - mkdirp(tmp_path) + tmp_path = tempfile.mkdtemp(dir='/tmp', prefix='spack') env.set('TEST_TMPDIR', tmp_path) env.set('TF_SYSTEM_LIBS', 'com_google_protobuf') @@ -743,11 +743,13 @@ class PyTensorflow(Package, CudaPackage): build_pip_package('--src', buildpath) def install(self, spec, prefix): + tmp_path = env['TEST_TMPDIR'] buildpath = join_path(self.stage.source_path, 'spack-build') with working_dir(buildpath): setup_py('install', '--prefix={0}'.format(prefix), '--single-version-externally-managed', '--root=/') + remove_linked_tree(tmp_path) @run_after('install') @on_package_attributes(run_tests=True) -- cgit v1.2.3-70-g09d2