[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
deepmodeling · Mar 18, 2024 · 4965529 · 4965529
1 parent 5cd0d56
commit 4965529
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepmd/pt/model/task/denoise.py b/deepmd/pt/model/task/denoise.py
@@ -474,7 +474,7 @@ def forward(
             )  # Shape is [nframes, natoms[0], net_dim_out]
             outs_logits = outs_logits + atom_logits
         # TODO:
-        '''
+        """
         else:
             for type_i, ll in enumerate(self.filter_layers_coord.networks):
                 mask = (atype == type_i).unsqueeze(-1)
@@ -493,7 +493,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