Skip to content

Commit

Permalink
Update make_base_descriptor.py
Browse files Browse the repository at this point in the history
  • Loading branch information
iProzd committed Feb 28, 2024
1 parent c9eb767 commit 00105c7
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion deepmd/dpmodel/descriptor/make_base_descriptor.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,10 @@
abstractmethod,
)
from typing import (
Callable,
List,
Optional,
Union,
)

from deepmd.common import (
Expand Down Expand Up @@ -84,7 +86,9 @@ def mixed_types(self) -> bool:
"""
pass

def compute_input_stats(self, merged: callable, path: Optional[DPPath] = None):
def compute_input_stats(
self, merged: Union[Callable, List], path: Optional[DPPath] = None
):
"""Update mean and stddev for descriptor elements."""
raise NotImplementedError

Expand Down

0 comments on commit 00105c7

Please sign in to comment.