fix super

deepmodeling · Jan 8, 2024 · 8550b68 · 8550b68
1 parent dbd86f8
commit 8550b68
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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