From 8989b8ad528c858756e67bd589a00b3cbf22a471 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Fri, 1 Nov 2024 00:59:09 -0400
Subject: [PATCH] Update deepmd/dpmodel/descriptor/repformers.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
---
 deepmd/dpmodel/descriptor/repformers.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/deepmd/dpmodel/descriptor/repformers.py b/deepmd/dpmodel/descriptor/repformers.py
index 29802b8b65..67b51bbbdd 100644
--- a/deepmd/dpmodel/descriptor/repformers.py
+++ b/deepmd/dpmodel/descriptor/repformers.py
@@ -516,14 +516,13 @@ def serialize(self):
             # variables
             "g2_embd": self.g2_embd.serialize(),
             "repformer_layers": [layer.serialize() for layer in self.layers],
-            "env_mat": EnvMat(self.rcut, self.rcut_smth).serialize(),
+            "env_mat": self.env_mat.serialize(),
             "@variables": {
                 "davg": to_numpy_array(self["davg"]),
                 "dstd": to_numpy_array(self["dstd"]),
             },
         }
 
-
 # translated by GPT and modified
 def get_residual(
     _dim: int,