From ea7bd98d06f3a5b49302478b91c06b277e78ed0e Mon Sep 17 00:00:00 2001
From: Anyang Peng <137014849+anyangml@users.noreply.github.com>
Date: Tue, 12 Nov 2024 10:27:31 +0800
Subject: [PATCH] fix: dpmodel zbl rcut sel

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

diff --git a/deepmd/dpmodel/model/model.py b/deepmd/dpmodel/model/model.py
index 6dea9041fc..8fa210e864 100644
--- 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"],
     )