annotated cast_precision temporarily and CDLL code for error

deepmodeling · Jan 10, 2024 · 70db55a · 70db55a
1 parent 7d90ef0
commit 70db55a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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