fix: add update_deepmd_input to tf unit test

Author: OutisLi

source/tests/tf/test_data_modifier.py

@@ -22,6 +22,9 @@
 from deepmd.utils.argcheck import (
     normalize,
 )
+from deepmd.utils.compat import (
+    update_deepmd_input,
+)
 
 from .common import (
     Data,
@@ -56,6 +59,7 @@ def _setUp(self) -> None:
             restart=None, init_model=None, log_path=None, log_level=30, mpi_log="master"
         )
         jdata = j_loader(INPUT)
+        jdata = update_deepmd_input(jdata, warning=True)
         jdata = normalize(jdata)
         # init model
         model = DPTrainer(jdata, run_opt=run_opt)

source/tests/tf/test_data_modifier_shuffle.py

@@ -26,6 +26,9 @@
 from deepmd.utils.argcheck import (
     normalize,
 )
+from deepmd.utils.compat import (
+    update_deepmd_input,
+)
 
 from ..seed import (
     GLOBAL_SEED,
@@ -61,6 +64,7 @@ def _setUp(self) -> None:
             restart=None, init_model=None, log_path=None, log_level=30, mpi_log="master"
         )
         jdata = self._setUp_jdata()
+        jdata = update_deepmd_input(jdata, warning=True)
         jdata = normalize(jdata)
         self._setUp_data()