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

Fixing local directory (#977)

Browse Source 
* Fixing local directory

Enables ability to set custom local directory; code may be messy.

* Create all intermediate parent directories
This commit is contained in:
Richard Liaw
2017-09-14 10:33:52 -07:00
committed by Philipp Moritz
parent 1eb8c83314
commit d516d9440e
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/common.py
+4 -1
View File
@@ -114,7 +114,10 @@ class Agent(object):
self.__class__.__name__,
datetime.today().strftime("%Y-%m-%d_%H-%M-%S"))
if upload_dir.startswith("file"):
self.logdir = tempfile.mkdtemp(prefix=prefix, dir="/tmp/ray")
local_dir = upload_dir[len("file://"):]
if not os.path.exists(local_dir):
os.makedirs(local_dir)
self.logdir = tempfile.mkdtemp(prefix=prefix, dir=local_dir)
else:
self.logdir = os.path.join(upload_dir, prefix)