fix: UT

deepmodeling · Oct 31, 2024 · 106de1b · 106de1b
1 parent a95d75e
commit 106de1b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/deepmd/dpmodel/model/model.py b/deepmd/dpmodel/model/model.py
@@ -11,6 +11,9 @@
 from deepmd.dpmodel.fitting.ener_fitting import (
     EnergyFittingNet,
 )
+from deepmd.dpmodel.descriptor.base_descriptor import (
+    BaseDescriptor,
+)
 from deepmd.dpmodel.model.base_model import (
     BaseModel,
 )
@@ -65,7 +68,7 @@ def get_standard_model(data: dict) -> EnergyModel:
 
 
 def get_zbl_model(data: dict):
-    descriptor = DescrptSeA(**data["descriptor"])
+    descriptor = BaseDescriptor(**data["descriptor"])
     fitting_type = data["fitting_net"].pop("type")
     if fitting_type == "ener":
         fitting = EnergyFittingNet(