serialize protection and use_exp_switch for EnvMat

njzjz · njzjz · commit fcdd130b4ea3 · 2025-05-21T23:10:16.000+08:00
Signed-off-by: Jinzhe Zeng &lt;jinzhe.zeng@ustc.edu.cn&gt;
diff --git a/deepmd/dpmodel/utils/env_mat.py b/deepmd/dpmodel/utils/env_mat.py
@@ -177,6 +177,8 @@ def serialize(
         return {
             "rcut": self.rcut,
             "rcut_smth": self.rcut_smth,
+            "protection": self.protection,
+            "use_exp_switch": self.use_exp_switch,
         }
 
     @classmethod

Original file line number	Diff line number	Diff line change
`@@ -177,6 +177,8 @@ def serialize(`
`177`	`177`	`return {`
`178`	`178`	`"rcut": self.rcut,`
`179`	`179`	`"rcut_smth": self.rcut_smth,`
	`180`	`+ "protection": self.protection,`
	`181`	`+ "use_exp_switch": self.use_exp_switch,`
`180`	`182`	`}`
`181`	`183`
`182`	`184`	`@classmethod`