Skip to content

Commit 5331422

Browse files
committed
fix: add argcheck normalizer to tf unit test
1 parent 64fa919 commit 5331422

2 files changed

Lines changed: 8 additions & 0 deletions

File tree

source/tests/tf/test_data_modifier.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,9 @@
1919
from deepmd.tf.utils.data_system import (
2020
DeepmdDataSystem,
2121
)
22+
from deepmd.utils.argcheck import (
23+
normalize,
24+
)
2225

2326
from .common import (
2427
Data,
@@ -53,6 +56,7 @@ def _setUp(self) -> None:
5356
restart=None, init_model=None, log_path=None, log_level=30, mpi_log="master"
5457
)
5558
jdata = j_loader(INPUT)
59+
jdata = normalize(jdata)
5660
# init model
5761
model = DPTrainer(jdata, run_opt=run_opt)
5862
rcut = model.model.get_rcut()

source/tests/tf/test_data_modifier_shuffle.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,9 @@
2323
from deepmd.tf.utils.data_system import (
2424
DeepmdDataSystem,
2525
)
26+
from deepmd.utils.argcheck import (
27+
normalize,
28+
)
2629

2730
from ..seed import (
2831
GLOBAL_SEED,
@@ -58,6 +61,7 @@ def _setUp(self) -> None:
5861
restart=None, init_model=None, log_path=None, log_level=30, mpi_log="master"
5962
)
6063
jdata = self._setUp_jdata()
64+
jdata = normalize(jdata)
6165
self._setUp_data()
6266

6367
# init model

0 commit comments

Comments
 (0)