From 31edbb1cb3a2f4e270f230120d54ee1ae80b8688 Mon Sep 17 00:00:00 2001
From: HydrogenSulfate <490868991@qq.com>
Date: Wed, 11 Sep 2024 10:55:24 +0800
Subject: [PATCH] fix for logpath is None

---
 deepmd/pt/entrypoints/main.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/deepmd/pt/entrypoints/main.py b/deepmd/pt/entrypoints/main.py
index 274a682f1..70f7206a0 100644
--- a/deepmd/pt/entrypoints/main.py
+++ b/deepmd/pt/entrypoints/main.py
@@ -555,7 +555,11 @@ def main(args: Optional[Union[List[str], argparse.Namespace]] = None):
     else:
         FLAGS = args
 
-    set_log_handles(FLAGS.log_level, Path(FLAGS.log_path), mpi_log=None)
+    set_log_handles(
+        FLAGS.log_level,
+        Path(args.log_path) if args.log_path else None,
+        mpi_log=None,
+    )
     log.debug("Log handles were successfully set")
     log.info("DeePMD version: %s", __version__)