fix pre-commit

deepmodeling · Mar 18, 2024 · cea7476 · cea7476
1 parent 0dc11d6
commit cea7476
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deepmd/pt/model/task/denoise.py b/deepmd/pt/model/task/denoise.py
@@ -293,7 +293,7 @@ def serialize(self) -> dict:
         }
 
     @classmethod
-    def deserialize(cls, data: dict) -> "DenoiseFitting":
+    def deserialize(cls, data: dict) -> "DenoiseFittingNet":
         data = copy.deepcopy(data)
         variables = data.pop("@variables")
         nets = data.pop("nets")
@@ -494,6 +494,7 @@ def forward(
                 outs_logits + atom_logits
             )
         # TODO:
+        '''
         else:
             for type_i, ll in enumerate(self.filter_layers_coord.networks):
                 mask = (atype == type_i).unsqueeze(-1)
@@ -512,6 +513,7 @@ def forward(
                 outs = (
                     outs + atom_property
                 )  # Shape is [nframes, natoms[0], net_dim_out]
+        '''
         # nf x nloc
         mask = self.emask(atype)
         # nf x nloc x nod