From d516d9440e3b99ab534601fa012ebff73d0b46ed Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Thu, 14 Sep 2017 10:33:52 -0700 Subject: [PATCH] Fixing local directory (#977) * Fixing local directory Enables ability to set custom local directory; code may be messy. * Create all intermediate parent directories --- python/ray/rllib/common.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/ray/rllib/common.py b/python/ray/rllib/common.py index b46987202..9c98bc52a 100644 --- a/python/ray/rllib/common.py +++ b/python/ray/rllib/common.py @@ -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)