fix ut

wanghan-iapcm · Apr 4, 2024 · 7ab0a4f · 7ab0a4f
1 parent 75591a1
commit 7ab0a4f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/deepmd/dpmodel/atomic_model/pairtab_atomic_model.py b/deepmd/dpmodel/atomic_model/pairtab_atomic_model.py
@@ -59,6 +59,8 @@ def __init__(
         rcut: float,
         sel: Union[int, List[int]],
         type_map: List[str],
+        rcond: Optional[float] = None,
+        atom_ener: Optional[List[float]] = None,
         **kwargs,
     ):
         super().__init__(type_map, **kwargs)
@@ -69,6 +71,8 @@ def __init__(
         self.tab = PairTab(self.tab_file, rcut=rcut)
         self.type_map = type_map
         self.ntypes = len(type_map)
+        self.rcond = rcond
+        self.atom_ener = atom_ener
 
         if self.tab_file is not None:
             self.tab_info, self.tab_data = self.tab.get()