fix for loop; fix codeql warnings

deepmodeling · Oct 31, 2024 · 15bf3e2 · 15bf3e2
1 parent cb15335
commit 15bf3e2
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/deepmd/pt/utils/tabulate.py b/deepmd/pt/utils/tabulate.py
@@ -299,6 +299,7 @@ def _get_descrpt_type(self):
             return "R"
         elif isinstance(self.descrpt, deepmd.pt.model.descriptor.DescrptSeT):
             return "T"
+        raise RuntimeError(f"Unsupported descriptor {self.descrpt}")
 
     def _get_layer_size(self):
         # get the number of layers in EmbeddingNet

diff --git a/deepmd/tf/utils/tabulate.py b/deepmd/tf/utils/tabulate.py
@@ -180,6 +180,7 @@ def _get_descrpt_type(self):
             return "T"
         elif isinstance(self.descrpt, deepmd.tf.descriptor.DescrptSeR):
             return "R"
+        raise RuntimeError(f"Unsupported descriptor {self.descrpt}")
 
     def _get_bias(self):
         bias = {}

diff --git a/deepmd/utils/tabulate.py b/deepmd/utils/tabulate.py
@@ -172,6 +172,12 @@ def build(
             ).astype(int)
             idx = 0
             for ii in range(self.ntypes):
+                if self.is_pt:
+                    uu = np.max(upper[ii])
+                    ll = np.min(lower[ii])
+                else:
+                    ll = lower[ii]
+                    uu = upper[ii]
                 for jj in range(ii, self.ntypes):
                     net = "filter_" + str(ii) + "_net_" + str(jj)
                     self._build_lower(