From 1cb163a02c0c063f13b1ccc618f041b513f9233d Mon Sep 17 00:00:00 2001 From: Han Wang Date: Fri, 15 Mar 2024 23:40:54 +0800 Subject: [PATCH] fix typo --- deepmd/dpmodel/atomic_model/base_atomic_model.py | 6 +++--- deepmd/pt/model/atomic_model/base_atomic_model.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/deepmd/dpmodel/atomic_model/base_atomic_model.py b/deepmd/dpmodel/atomic_model/base_atomic_model.py index dd0733abd0..42d1e67138 100644 --- a/deepmd/dpmodel/atomic_model/base_atomic_model.py +++ b/deepmd/dpmodel/atomic_model/base_atomic_model.py @@ -105,9 +105,9 @@ def forward_common_atomic( ret_dict dict of output atomic properties. should implement the definition of `fitting_output_def`. - ret_dit["mask"] of shape nf x nloc will be provided. - ret_dit["mask"][ff,ii] == 1 indicating the ii-th atom of the ff-th frame is real. - ret_dit["mask"][ff,ii] == 0 indicating the ii-th atom of the ff-th frame is virtual. + ret_dict["mask"] of shape nf x nloc will be provided. + ret_dict["mask"][ff,ii] == 1 indicating the ii-th atom of the ff-th frame is real. + ret_dict["mask"][ff,ii] == 0 indicating the ii-th atom of the ff-th frame is virtual. """ _, nloc, _ = nlist.shape diff --git a/deepmd/pt/model/atomic_model/base_atomic_model.py b/deepmd/pt/model/atomic_model/base_atomic_model.py index 432d9b2606..997e6fe699 100644 --- a/deepmd/pt/model/atomic_model/base_atomic_model.py +++ b/deepmd/pt/model/atomic_model/base_atomic_model.py @@ -132,9 +132,9 @@ def forward_common_atomic( ret_dict dict of output atomic properties. should implement the definition of `fitting_output_def`. - ret_dit["mask"] of shape nf x nloc will be provided. - ret_dit["mask"][ff,ii] == 1 indicating the ii-th atom of the ff-th frame is real. - ret_dit["mask"][ff,ii] == 0 indicating the ii-th atom of the ff-th frame is virtual. + ret_dict["mask"] of shape nf x nloc will be provided. + ret_dict["mask"][ff,ii] == 1 indicating the ii-th atom of the ff-th frame is real. + ret_dict["mask"][ff,ii] == 0 indicating the ii-th atom of the ff-th frame is virtual. """ _, nloc, _ = nlist.shape