diff --git a/deepmd/tf/descriptor/se_r.py b/deepmd/tf/descriptor/se_r.py index c34734a8cf..ba24142987 100644 --- a/deepmd/tf/descriptor/se_r.py +++ b/deepmd/tf/descriptor/se_r.py @@ -765,9 +765,6 @@ def serialize(self, suffix: str = "") -> dict: raise NotImplementedError("spin is unsupported") assert self.davg is not None assert self.dstd is not None - # TODO: tf: handle type embedding in DescrptSeR.serialize - # not sure how to handle type embedding - type embedding is not a model parameter, - # but instead a part of the input data. Maybe the interface should be refactored... return { "@class": "Descriptor", "type": "se_r",