set is_pt

Author: cherryWangY

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",

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

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