fix folder name

calico · Jun 20, 2024 · 7116151 · 7116151
1 parent a4ce96c
commit 7116151
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/baskerville/trainer.py b/src/baskerville/trainer.py
@@ -122,8 +122,8 @@ def __init__(
 
         # if log_dir is in gcs then create a local temp dir
         if is_gcs_path(self.log_dir):
-            folder_name = self.log_dir.split("/")[-1]
-            self.log_dir = tempfile.mkdtemp() + folder_name
+            folder_name = "/".join(self.log_dir.split("/")[3:])
+            self.log_dir = tempfile.mkdtemp() + "/" + folder_name
             self.gcs_log_dir = log_dir
             self.gcs = True
         else: