Add config option for pre-processing

ACEsuit · Oct 30, 2024 · d7c80e1 · d7c80e1
1 parent f6124f2
commit d7c80e1
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/mace/tools/arg_parser.py b/mace/tools/arg_parser.py
@@ -700,7 +700,21 @@ def build_default_arg_parser() -> argparse.ArgumentParser:
 
 
 def build_preprocess_arg_parser() -> argparse.ArgumentParser:
-    parser = argparse.ArgumentParser()
+    try:
+        import configargparse
+
+        parser = configargparse.ArgumentParser(
+            config_file_parser_class=configargparse.YAMLConfigFileParser,
+        )
+        parser.add(
+            "--config",
+            type=str,
+            is_config_file=True,
+            help="config file to agregate options",
+        )
+    except ImportError:
+        parser = argparse.ArgumentParser()
+
     parser.add_argument(
         "--train_file",
         help="Training set h5 file",