Skip to content

Commit 400d19a

Browse files
Copilotnjzjz
andcommitted
fix(dpmodel): correct return type annotations in dpa2 and hybrid descriptors
Co-authored-by: njzjz <9496702+njzjz@users.noreply.github.com>
1 parent 149bee7 commit 400d19a

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

deepmd/dpmodel/descriptor/dpa2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -788,7 +788,7 @@ def set_stat_mean_and_stddev(
788788

789789
def get_stat_mean_and_stddev(
790790
self,
791-
) -> tuple[tuple[Array, Array], list[Array]]:
791+
) -> tuple[list[Array], list[Array]]:
792792
"""Get mean and stddev for descriptor."""
793793
mean_list = [self.repinit.mean, self.repformers.mean]
794794
stddev_list = [

deepmd/dpmodel/descriptor/hybrid.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -214,7 +214,7 @@ def set_stat_mean_and_stddev(
214214
def get_stat_mean_and_stddev(
215215
self,
216216
) -> tuple[
217-
list[tuple[Array, Array]],
217+
list[Union[Array, list[Array]]],
218218
list[Union[Array, list[Array]]],
219219
]:
220220
"""Get mean and stddev for descriptor."""

0 commit comments

Comments
 (0)