diff --git a/deepmd/descriptor/se_a.py b/deepmd/descriptor/se_a.py index 73221c1f91..d964cf824a 100644 --- a/deepmd/descriptor/se_a.py +++ b/deepmd/descriptor/se_a.py @@ -273,7 +273,9 @@ def get_dim_rot_mat_1(self) -> int: return self.filter_neuron[-1] def get_nlist(self) -> Tuple[tf.Tensor, tf.Tensor, List[int], List[int]]: - """Returns + """Returns neighbor information. + + Returns ------- nlist Neighbor list diff --git a/deepmd/descriptor/se_a_ef.py b/deepmd/descriptor/se_a_ef.py index a94939f9db..0dbad403f4 100644 --- a/deepmd/descriptor/se_a_ef.py +++ b/deepmd/descriptor/se_a_ef.py @@ -140,7 +140,9 @@ def get_rot_mat(self) -> tf.Tensor: return self.qmat def get_nlist(self) -> Tuple[tf.Tensor, tf.Tensor, List[int], List[int]]: - """Returns + """Returns neighbor information. + + Returns ------- nlist Neighbor list diff --git a/deepmd/descriptor/se_r.py b/deepmd/descriptor/se_r.py index 664848d248..3ed02ec8ce 100644 --- a/deepmd/descriptor/se_r.py +++ b/deepmd/descriptor/se_r.py @@ -190,7 +190,9 @@ def get_dim_out(self): return self.filter_neuron[-1] def get_nlist(self): - """Returns + """Returns neighbor information. + + Returns ------- nlist Neighbor list diff --git a/deepmd/descriptor/se_t.py b/deepmd/descriptor/se_t.py index f908fd2e32..7615275a82 100644 --- a/deepmd/descriptor/se_t.py +++ b/deepmd/descriptor/se_t.py @@ -193,7 +193,9 @@ def get_dim_out(self) -> int: return self.filter_neuron[-1] def get_nlist(self) -> Tuple[tf.Tensor, tf.Tensor, List[int], List[int]]: - """Returns + """Returns neighbor information. + + Returns ------- nlist Neighbor list