From 06c9e1176f99e3f6c17a5c770784995181e5ed22 Mon Sep 17 00:00:00 2001
From: Anyang Peng <137014849+anyangml@users.noreply.github.com>
Date: Thu, 31 Oct 2024 20:56:05 +0800
Subject: [PATCH] fix: UT

---
 deepmd/dpmodel/model/dp_zbl_model.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/deepmd/dpmodel/model/dp_zbl_model.py b/deepmd/dpmodel/model/dp_zbl_model.py
index 4c6fec281f..561277ed95 100644
--- a/deepmd/dpmodel/model/dp_zbl_model.py
+++ b/deepmd/dpmodel/model/dp_zbl_model.py
@@ -20,17 +20,18 @@
     make_model,
 )
 
-DPZBLEnergyModel_ = make_model(DPZBLLinearEnergyAtomicModel)
+DPZBLModel_ = make_model(DPZBLLinearEnergyAtomicModel)
 
 
 @BaseModel.register("zbl")
-class DPZBLModel(DPZBLEnergyModel_):
+class DPZBLModel(DPZBLModel_):
+    model_type = "ener"
     def __init__(
         self,
         *args,
         **kwargs,
     ):
-        DPZBLEnergyModel_.__init__(self, *args, **kwargs)
+        super().__init__(*args, **kwargs)
 
 
 @classmethod