feat: logdir-gitignore (#527)

* logdir-gitignore --------- Co-authored-by: KAAANG <[email protected]>
SwanHubX · May 1, 2024 · ee52389 · ee52389
1 parent d2d0942
commit ee52389
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/swanlab/data/sdk.py b/swanlab/data/sdk.py
@@ -393,6 +393,10 @@ def _init_logdir(logdir: str) -> str:
                 raise IOError
         except IOError:
             raise IOError("logdir must have Write permission.")
+    # 如果logdir是空的，创建.gitignore文件，写入*
+    if not os.listdir(logdir):
+        with open(os.path.join(logdir, ".gitignore"), "w") as f:
+            f.write("*")
     return logdir