experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make sure temp directory on windows has the correct path separator.

Browse Source 
PiperOrigin-RevId: 225462105
This commit is contained in:
Gunhan Gulsoy 2018-12-13 17:06:32 -08:00 committed by TensorFlower Gardener
parent 97fe55e68b
commit 2a67515699
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/platform/googletest.py
View File

@ -112,6 +112,9 @@ def GetTempDir():
os.path.basename(tf_inspect.getfile(first_frame))) os.path.basename(tf_inspect.getfile(first_frame)))
temp_dir = tempfile.mkdtemp(prefix=temp_dir.rstrip('.py')) temp_dir = tempfile.mkdtemp(prefix=temp_dir.rstrip('.py'))
# Make sure we have the correct path separators.
temp_dir = temp_dir.replace('/', os.sep)
def delete_temp_dir(dirname=temp_dir): def delete_temp_dir(dirname=temp_dir):
try: try:
file_io.delete_recursively(dirname) file_io.delete_recursively(dirname)
@ -119,6 +122,7 @@ def GetTempDir():
logging.error('Error removing %s: %s', dirname, e) logging.error('Error removing %s: %s', dirname, e)
atexit.register(delete_temp_dir) atexit.register(delete_temp_dir)
_googletest_temp_dir = temp_dir _googletest_temp_dir = temp_dir
return _googletest_temp_dir return _googletest_temp_dir