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)