Update main.py

Signed-off-by: HydrogenSulfate <[email protected]>
HydrogenSulfate · Sep 11, 2024 · 5466939 · 5466939
1 parent 31edbb1
commit 5466939
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")