Skip to content

Commit 680412a

Browse files
[Feature] make csv logger exist_ok on logging folder (#1561)
Signed-off-by: Matteo Bettini <matbet@meta.com>
1 parent fc9794d commit 680412a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

torchrl/record/loggers/csv.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,10 @@ def __init__(self, log_dir: str):
2121
self.videos_counter = defaultdict(lambda: 0)
2222
self.text_counter = defaultdict(lambda: 0)
2323
self.log_dir = log_dir
24-
os.makedirs(self.log_dir)
25-
os.makedirs(os.path.join(self.log_dir, "scalars"))
26-
os.makedirs(os.path.join(self.log_dir, "videos"))
27-
os.makedirs(os.path.join(self.log_dir, "texts"))
24+
os.makedirs(self.log_dir, exist_ok=True)
25+
os.makedirs(os.path.join(self.log_dir, "scalars"), exist_ok=True)
26+
os.makedirs(os.path.join(self.log_dir, "videos"), exist_ok=True)
27+
os.makedirs(os.path.join(self.log_dir, "texts"), exist_ok=True)
2828

2929
self.files = {}
3030

0 commit comments

Comments
 (0)