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 (