diff options
Diffstat (limited to 'tensorflow/python/platform/googletest.py')
-rw-r--r-- | tensorflow/python/platform/googletest.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/platform/googletest.py b/tensorflow/python/platform/googletest.py index 5227d2e35c..1e74b1512b 100644 --- a/tensorflow/python/platform/googletest.py +++ b/tensorflow/python/platform/googletest.py @@ -99,6 +99,7 @@ def main(argv=None): # pylint: disable=function-redefined def GetTempDir(): """Return a temporary directory for tests to use.""" + global _googletest_temp_dir if not _googletest_temp_dir: first_frame = inspect.stack()[-1][0] temp_dir = os.path.join( @@ -112,7 +113,6 @@ def GetTempDir(): logging.error('Error removing %s: %s', dirname, e) atexit.register(delete_temp_dir) - global _googletest_temp_dir _googletest_temp_dir = temp_dir return _googletest_temp_dir |