fix: remove --input_script from dp test

`--input_script` is not used in the current code. Also, mark `--rand-seed` and `--atomic` is supported in all backends. Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · May 21, 2024 · 8e58d9a · 8e58d9a
1 parent 81b5949
commit 8e58d9a
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/deepmd/main.py b/deepmd/main.py
@@ -383,7 +383,7 @@ def main_parser() -> argparse.ArgumentParser:
         "--rand-seed",
         type=int,
         default=None,
-        help="(Supported backend: TensorFlow) The random seed",
+        help="The random seed",
     )
     parser_tst.add_argument(
         "--shuffle-test", action="store_true", default=False, help="Shuffle test data"
@@ -400,13 +400,7 @@ def main_parser() -> argparse.ArgumentParser:
         "--atomic",
         action="store_true",
         default=False,
-        help="(Supported backend: TensorFlow) Test the accuracy of atomic label, i.e. energy / tensor (dipole, polar)",
-    )
-    parser_tst.add_argument(
-        "-i",
-        "--input_script",
-        type=str,
-        help="(Supported backend: PyTorch) The input script of the model",
+        help="Test the accuracy of atomic label, i.e. energy / tensor (dipole, polar)",
     )
     parser_tst.add_argument(
         "--head",