Skip to content

Commit

Permalink
Solve alert
Browse files Browse the repository at this point in the history
  • Loading branch information
Chengqian-Zhang committed Jun 4, 2024
1 parent 45f9755 commit 9d4d0e3
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 19 deletions.
2 changes: 1 addition & 1 deletion deepmd/pt/model/descriptor/se_atten_v2.py
Original file line number Diff line number Diff line change
Expand Up @@ -231,7 +231,7 @@ def deserialize(cls, data: dict) -> "DescrptDPA1":
embeddings = data.pop("embeddings")
type_embedding = data.pop("type_embedding")
attention_layers = data.pop("attention_layers")
env_mat = data.pop("env_mat")
data.pop("env_mat")
embeddings_strip = data.pop("embeddings_strip")
obj = cls(**data)

Expand Down
18 changes: 0 additions & 18 deletions source/tests/consistent/descriptor/test_se_atten_v2.py
Original file line number Diff line number Diff line change
Expand Up @@ -209,24 +209,6 @@ def setUp(self):
dtype=GLOBAL_NP_FLOAT_PRECISION,
)
self.natoms = np.array([6, 6, 2, 4], dtype=np.int32)
(
tebd_dim,
resnet_dt,
type_one_side,
attn,
attn_layer,
attn_dotr,
excluded_types,
env_protection,
set_davg_zero,
scaling_factor,
normalize,
temperature,
ln_eps,
concat_output_tebd,
precision,
use_econf_tebd,
) = self.param

def build_tf(self, obj: Any, suffix: str) -> Tuple[list, dict]:
return self.build_tf_descriptor(
Expand Down

0 comments on commit 9d4d0e3

Please sign in to comment.