[REF] Use exist_ok=True

f-dangel · Aug 29, 2024 · 59bf40e · 59bf40e
1 parent f1e4f81
commit 59bf40e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/wandb_preempt/checkpointer.py b/wandb_preempt/checkpointer.py
@@ -151,8 +151,7 @@ def checkpoint_path(self, epoch: int) -> str:
             The path to the checkpoint file for this epoch.
         """
         checkpoint_dir = path.join(self.savedir, environ["SLURM_ARRAY_JOB_ID"])
-        if not path.exists(checkpoint_dir):
-            makedirs(checkpoint_dir)
+        makedirs(checkpoint_dir, exist_ok=True)
         return path.join(checkpoint_dir, f"{self.run_id}_epoch_{epoch:08g}.pt")
 
     def save_checkpoint(self, epoch: int) -> None: