diff --git a/deepmd_utils/model_format/network.py b/deepmd_utils/model_format/network.py index dda6a68c55..04aaa75534 100644 --- a/deepmd_utils/model_format/network.py +++ b/deepmd_utils/model_format/network.py @@ -354,7 +354,7 @@ def __init__( ).serialize() ) i_in = i_ot - super(EmbeddingNet, self).__init__(layers) + super().__init__(layers) self.in_dim = in_dim self.neuron = neuron self.activation_function = activation_function @@ -387,5 +387,5 @@ def deserialize(cls, data: dict) -> "EmbeddingNet": """ layers = data.pop("layers") obj = cls(**data) - super().__init__(layers) + super(EmbeddingNet, obj).__init__(layers) return obj