From 49655298c2ab646cd374e40ed04b4bec749fe700 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 18 Mar 2024 16:14:35 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- deepmd/pt/model/task/denoise.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deepmd/pt/model/task/denoise.py b/deepmd/pt/model/task/denoise.py index da1f78b4db..48d6893f8b 100644 --- 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