update data modifier in unit tests

Author: ChiahsinChu

source/tests/tf/test_data_modifier_shuffle.py

@@ -8,12 +8,12 @@
     GLOBAL_NP_FLOAT_PRECISION,
     tf,
 )
-from deepmd.tf.infer.data_modifier import (
-    DipoleChargeModifier,
-)
 from deepmd.tf.infer.deep_dipole import (
     DeepDipole,
 )
+from deepmd.tf.modifier import (
+    DipoleChargeModifier,
+)
 from deepmd.tf.train.run_options import (
     RunOptions,
 )

source/tests/tf/test_dipolecharge.py

@@ -7,7 +7,7 @@
 from deepmd.tf.env import (
     GLOBAL_NP_FLOAT_PRECISION,
 )
-from deepmd.tf.infer import (
+from deepmd.tf.modifier import (
     DipoleChargeModifier,
 )
 from deepmd.tf.utils.convert import (
@@ -32,7 +32,11 @@ def setUpClass(cls) -> None:
             "dipolecharge_d.pb",
         )
         cls.dp = DipoleChargeModifier(
-            "dipolecharge_d.pb", [-1.0, -3.0], [1.0, 1.0, 1.0, 1.0, 1.0], 4.0, 0.2
+            model_name="dipolecharge_d.pb",
+            model_charge_map=[-1.0, -3.0],
+            sys_charge_map=[1.0, 1.0, 1.0, 1.0, 1.0],
+            ewald_h=4.0,
+            ewald_beta=0.2,
         )
 
     def setUp(self) -> None: