fix: dpmodel zbl rcut sel

anyangml · Nov 12, 2024 · ea7bd98 · ea7bd98
1 parent 3a95d22
commit ea7bd98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepmd/dpmodel/model/model.py b/deepmd/dpmodel/model/model.py
@@ -80,8 +80,8 @@ def get_zbl_model(data: dict) -> DPZBLModel:
     filepath = data["use_srtab"]
     pt_model = PairTabAtomicModel(
         filepath,
-        data["descriptor"]["rcut"],
-        data["descriptor"]["sel"],
+        descriptor.get_rcut(),
+        descriptor.get_sel(),
         type_map=data["type_map"],
     )