From ab36dbabfb53362eaa6770476fa2c3a23636325e Mon Sep 17 00:00:00 2001
From: Han Wang <wang_han@iapcm.ac.cn>
Date: Wed, 10 Jan 2024 14:32:32 +0800
Subject: [PATCH] fix interface of abs class base Descriptor

---
 deepmd_pt/model/descriptor/descriptor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepmd_pt/model/descriptor/descriptor.py b/deepmd_pt/model/descriptor/descriptor.py
index fecb86e..2d20733 100644
--- a/deepmd_pt/model/descriptor/descriptor.py
+++ b/deepmd_pt/model/descriptor/descriptor.py
@@ -84,9 +84,9 @@ def init_desc_stat(self, sumr, suma, sumn, sumr2, suma2):
     @abstractmethod
     def forward(
         self,
-        nlist,
         extended_coord,
         extended_atype,
+        nlist,
         mapping: Optional[torch.Tensor] = None,
     ):
         """Calculate descriptor.