From eb14b3b39a23ba6e7a4463cd6d8b04e7399598e7 Mon Sep 17 00:00:00 2001
From: Shane A <shanea@allenai.org>
Date: Tue, 10 Dec 2024 14:06:44 -0800
Subject: [PATCH] Pass dir as str to wandb.init

---
 scripts/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/train.py b/scripts/train.py
index b4d89be2d..8d7d85a2a 100644
--- a/scripts/train.py
+++ b/scripts/train.py
@@ -107,7 +107,7 @@ def main(cfg: TrainConfig) -> None:
         wandb_dir = Path(cfg.save_folder) / "wandb"
         wandb_dir.mkdir(parents=True, exist_ok=True)
         wandb.init(
-            dir=wandb_dir,
+            dir=str(wandb_dir),
             project=cfg.wandb.project,
             entity=cfg.wandb.entity,
             group=cfg.wandb.group,