From 546693947e059463daf2d378b371e493f0e0d95f Mon Sep 17 00:00:00 2001
From: HydrogenSulfate <490868991@qq.com>
Date: Thu, 12 Sep 2024 01:52:04 +0800
Subject: [PATCH] Update main.py

Signed-off-by: HydrogenSulfate <490868991@qq.com>
---
 deepmd/pt/entrypoints/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepmd/pt/entrypoints/main.py b/deepmd/pt/entrypoints/main.py
index 70f7206a0f..03b9f8c3a1 100644
--- a/deepmd/pt/entrypoints/main.py
+++ b/deepmd/pt/entrypoints/main.py
@@ -557,7 +557,7 @@ def main(args: Optional[Union[List[str], argparse.Namespace]] = None):
 
     set_log_handles(
         FLAGS.log_level,
-        Path(args.log_path) if args.log_path else None,
+        Path(FLAGS.log_path) if FLAGS.log_path else None,
         mpi_log=None,
     )
     log.debug("Log handles were successfully set")