Source code for deepmd.pt.model.atomic_model.dos_atomic_model
# SPDX-License-Identifier: LGPL-3.0-or-later
from deepmd.pt.model.task.dos import (
DOSFittingNet,
)
from .dp_atomic_model import (
DPAtomicModel,
)
[docs]
class DPDOSAtomicModel(DPAtomicModel):
def __init__(self, descriptor, fitting, type_map, **kwargs):
assert isinstance(fitting, DOSFittingNet)
super().__init__(descriptor, fitting, type_map, **kwargs)