From 050edd4c83f1af7dbf83aeb51e141bb94624c769 Mon Sep 17 00:00:00 2001
From: Anyang Peng <137014849+anyangml@users.noreply.github.com>
Date: Mon, 11 Nov 2024 11:59:11 +0800
Subject: [PATCH] fix: zbl rcut, sel

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

diff --git a/deepmd/pt/model/model/__init__.py b/deepmd/pt/model/model/__init__.py
index f3896ec853..03896f3e59 100644
--- a/deepmd/pt/model/model/__init__.py
+++ b/deepmd/pt/model/model/__init__.py
@@ -180,8 +180,8 @@ def get_zbl_model(model_params):
     filepath = model_params["use_srtab"]
     pt_model = PairTabAtomicModel(
         filepath,
-        model_params["descriptor"]["rcut"],
-        model_params["descriptor"]["sel"],
+        descriptor.get_rcut(),
+        descriptor.get_sel(),
         type_map=model_params["type_map"],
     )