set is_pt

deepmodeling · Oct 31, 2024 · cb15335 · cb15335
1 parent 39d6d17
commit cb15335
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/deepmd/pt/utils/tabulate.py b/deepmd/pt/utils/tabulate.py
@@ -59,9 +59,9 @@ def __init__(
             neuron,
             type_one_side,
             exclude_types,
+            True,
         )
         self.descrpt_type = self._get_descrpt_type()
-        self.is_pt = True
 
         supported_descrpt_type = (
             "Atten",

diff --git a/deepmd/tf/utils/tabulate.py b/deepmd/tf/utils/tabulate.py
@@ -78,10 +78,10 @@ def __init__(
             neuron,
             type_one_side,
             exclude_types,
+            False,
         )
 
         self.descrpt_type = self._get_descrpt_type()
-        self.is_pt = False
         """Constructor."""
         self.graph = graph
         self.graph_def = graph_def

diff --git a/deepmd/utils/tabulate.py b/deepmd/utils/tabulate.py
@@ -25,6 +25,7 @@ def __init__(
         neuron,
         type_one_side,
         exclude_types,
+        is_pt,
     ) -> None:
         """Constructor."""
         super().__init__()
@@ -34,10 +35,10 @@ def __init__(
         self.neuron = neuron
         self.type_one_side = type_one_side
         self.exclude_types = exclude_types
+        self.is_pt = is_pt
 
         """Need to be initialized in the subclass."""
         self.descrpt_type = "Base"
-        self.is_pt = True
 
         self.sel_a = []
         self.rcut = 0.0