diff options
| author |  Jonathan Hseu <jhseu@google.com> | 2017-03-27 18:12:33 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-27 21:05:24 -0700 |
| commit | 832f9a79af2040c3f190adecd6642c6b979b0ab3 (patch) | |
| tree | 244737bf02769df6b758610efd87be2323d029ae | |
| parent | ad93dc71f1154c0a4935b1bd86653f29acc8fdfa (diff) | |
Create a new tempdir for each test to fix the race condition for sharded tests.
Change: 151403279
| -rw-r--r-- | tensorflow/python/platform/googletest.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/tensorflow/python/platform/googletest.py b/tensorflow/python/platform/googletest.py index 456c1fd2ab..5227d2e35c 100644 --- a/tensorflow/python/platform/googletest.py +++ b/tensorflow/python/platform/googletest.py @@ -103,10 +103,7 @@ def GetTempDir(): first_frame = inspect.stack()[-1][0] temp_dir = os.path.join( tempfile.gettempdir(), os.path.basename(inspect.getfile(first_frame))) - temp_dir = temp_dir.rstrip('.py') - if not os.path.isdir(temp_dir): - os.mkdir(temp_dir, 0o755) - temp_dir = tempfile.mkdtemp(dir=temp_dir) + temp_dir = tempfile.mkdtemp(prefix=temp_dir.rstrip('.py')) def delete_temp_dir(dirname=temp_dir): try: |