fix: import

deepmodeling · Feb 2, 2024 · 1a59917 · 1a59917
1 parent 39da832
commit 1a59917
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/deepmd/pt/model/model/pair_tab_model.py b/deepmd/pt/model/model/pair_tab_model.py
@@ -11,20 +11,20 @@
     nn,
 )
 
-from deepmd.model_format import (
+from deepmd.dpmodel import (
     FittingOutputDef,
     OutputVariableDef,
 )
 from deepmd.utils.pair_tab import (
     PairTab,
 )
 
-from .atomic_model import (
-    AtomicModel,
+from .base_atomic_model import (
+    BaseAtomicModel,
 )
 
 
-class PairTabModel(nn.Module, AtomicModel):
+class PairTabModel(nn.Module, BaseAtomicModel):
     """Pairwise tabulation energy model.
 
     This model can be used to tabulate the pairwise energy between atoms for either