Skip to content

Commit 282c2a0

Browse files
authored
breaking: dp test all data (#2756)
Previously, we need to set `numb_test` to a quite large number to test all data. In this PR, passing 0 will lead to infinity. Breaking change: the default value of `numb_test` is set to 0. --------- Signed-off-by: Jinzhe Zeng <[email protected]>
1 parent f8bb170 commit 282c2a0

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

deepmd/entrypoints/test.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def test(
7171
set_prefix : str
7272
string prefix of set
7373
numb_test : int
74-
munber of tests to do
74+
munber of tests to do. 0 means all data.
7575
rand_seed : Optional[int]
7676
seed for random generator
7777
shuffle_test : bool
@@ -88,6 +88,9 @@ def test(
8888
RuntimeError
8989
if no valid system was found
9090
"""
91+
if numb_test == 0:
92+
# only float has inf, but should work for min
93+
numb_test = float("inf")
9194
if datafile is not None:
9295
datalist = open(datafile)
9396
all_sys = datalist.read().splitlines()

deepmd_cli/main.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,11 @@ def main_parser() -> argparse.ArgumentParser:
274274
"-S", "--set-prefix", default="set", type=str, help="The set prefix"
275275
)
276276
parser_tst.add_argument(
277-
"-n", "--numb-test", default=100, type=int, help="The number of data for test"
277+
"-n",
278+
"--numb-test",
279+
default=0,
280+
type=int,
281+
help="The number of data for test. 0 means all data.",
278282
)
279283
parser_tst.add_argument(
280284
"-r", "--rand-seed", type=int, default=None, help="The random seed"

0 commit comments

Comments
 (0)