From b428fc29461075247cf296626160961f99957a79 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Thu, 1 Feb 2024 08:24:38 -0500 Subject: [PATCH] fix cycle import Signed-off-by: Jinzhe Zeng --- deepmd/tf/infer/deep_eval.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/deepmd/tf/infer/deep_eval.py b/deepmd/tf/infer/deep_eval.py index 1a917d4f64..93417a10b3 100644 --- a/deepmd/tf/infer/deep_eval.py +++ b/deepmd/tf/infer/deep_eval.py @@ -45,9 +45,6 @@ default_tf_session_config, tf, ) -from deepmd.tf.infer.data_modifier import ( - DipoleChargeModifier, -) from deepmd.tf.utils.batch_size import ( AutoBatchSize, ) @@ -136,6 +133,10 @@ def __init__( # looks ugly... if self.modifier_type == "dipole_charge": + from deepmd.tf.infer.data_modifier import ( + DipoleChargeModifier, + ) + t_mdl_name = self._get_tensor("modifier_attr/mdl_name:0") t_mdl_charge_map = self._get_tensor("modifier_attr/mdl_charge_map:0") t_sys_charge_map = self._get_tensor("modifier_attr/sys_charge_map:0")