Source code for deepmd.pt.utils.update_sel

# SPDX-License-Identifier: LGPL-3.0-or-later
from typing import (
    Type,
)

from deepmd.pt.utils.neighbor_stat import (
    NeighborStat,
)
from deepmd.utils.update_sel import (
    BaseUpdateSel,
)



[docs]
class UpdateSel(BaseUpdateSel):
    @property

[docs]
    def neighbor_stat(self) -> Type[NeighborStat]:
        return NeighborStat



[docs]
    def hook(self, min_nbor_dist, max_nbor_size):
        # TODO: save to the model in UpdateSel.hook
        pass