wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-08 14:56:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Change /tmp to platform-specific temporary directory (#7529)

Browse Source
This commit is contained in:
mehrdadn
2020-03-16 18:10:14 -07:00
committed by GitHub
parent 797e6cfc2a
commit a0700e2f86
27 changed files with 289 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/examples/durable_trainable_example.py
+4 -2
View File
@@ -19,9 +19,11 @@ class MockDurableTrainable(DurableTrainable):
def __init__(self, remote_checkpoint_dir, *args, **kwargs):
# Mock the path as a local path.
local_dir_suffix = remote_checkpoint_dir.split("://")[1]
remote_checkpoint_dir = os.path.join("/tmp", local_dir_suffix)
remote_checkpoint_dir = os.path.join(ray.utils.get_user_temp_dir(),
local_dir_suffix)
# Disallow malformed relative paths for delete safety.
assert os.path.abspath(remote_checkpoint_dir).startswith("/tmp")
assert os.path.abspath(remote_checkpoint_dir).startswith(
ray.utils.get_user_temp_dir())
logger.info("Using %s as the mocked remote checkpoint directory.",
self.remote_checkpoint_dir)
super(MockDurableTrainable, self).__init__(remote_checkpoint_dir,