diff --git a/deepmd/entrypoints/test.py b/deepmd/entrypoints/test.py index 182b5a6eba..e348318f41 100644 --- a/deepmd/entrypoints/test.py +++ b/deepmd/entrypoints/test.py @@ -71,7 +71,7 @@ def test( set_prefix : str string prefix of set numb_test : int - munber of tests to do + munber of tests to do. 0 means all data. rand_seed : Optional[int] seed for random generator shuffle_test : bool @@ -88,6 +88,9 @@ def test( RuntimeError if no valid system was found """ + if numb_test == 0: + # only float has inf, but should work for min + numb_test = float("inf") if datafile is not None: datalist = open(datafile) all_sys = datalist.read().splitlines() diff --git a/deepmd_cli/main.py b/deepmd_cli/main.py index f4c28887bd..19cee642cf 100644 --- a/deepmd_cli/main.py +++ b/deepmd_cli/main.py @@ -274,7 +274,11 @@ def main_parser() -> argparse.ArgumentParser: "-S", "--set-prefix", default="set", type=str, help="The set prefix" ) parser_tst.add_argument( - "-n", "--numb-test", default=100, type=int, help="The number of data for test" + "-n", + "--numb-test", + default=0, + type=int, + help="The number of data for test. 0 means all data.", ) parser_tst.add_argument( "-r", "--rand-seed", type=int, default=None, help="The random seed"