From 70db55ab82b599328cc06239b63066d2ea04ea22 Mon Sep 17 00:00:00 2001 From: HydrogenSulfate <490868991@qq.com> Date: Wed, 10 Jan 2024 13:12:59 +0000 Subject: [PATCH] annotated cast_precision temporarily and CDLL code for error --- deepmd/descriptor/se_a.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/deepmd/descriptor/se_a.py b/deepmd/descriptor/se_a.py index 3233fe0211..b75b4cac2f 100644 --- a/deepmd/descriptor/se_a.py +++ b/deepmd/descriptor/se_a.py @@ -1026,8 +1026,12 @@ def _filter( outputs_size = [1, *self.filter_neuron] outputs_size_2 = self.n_axis_neuron # 16 all_excluded = all( - (type_input, type_i) in self.exclude_types # set() - for type_i in range(self.ntypes) + # FIXME: the bracket '[]' is needed when convert to static model, will be + # removed when fixed. + [ # noqa + (type_input, type_i) in self.exclude_types # set() noqa + for type_i in range(self.ntypes) + ] ) if all_excluded: # all types are excluded so result and qmat should be zeros