We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 93e6851 commit cbaf7fdCopy full SHA for cbaf7fd
1 file changed
deepmd/pt/model/descriptor/repflows.py
@@ -377,8 +377,6 @@ def forward(
377
mapping: Optional[torch.Tensor] = None,
378
comm_dict: Optional[dict[str, torch.Tensor]] = None,
379
):
380
- if comm_dict is None:
381
- assert mapping is not None
382
nframes, nloc, nnei = nlist.shape
383
nall = extended_coord.view(nframes, -1).shape[1] // 3
384
atype = extended_atype[:, :nloc]
0 commit comments