fix import bugs

wanghan-iapcm · Feb 13, 2024 · 5707074 · 5707074
1 parent 820d9f6
commit 5707074
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 6 deletions.
diff --git a/deepmd/dpmodel/__init__.py b/deepmd/dpmodel/__init__.py
@@ -1,7 +1,4 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
-from .atomic_model import (
-    DPAtomicModel,
-)
 from .common import (
     DEFAULT_PRECISION,
     PRECISION_DICT,
@@ -22,7 +19,6 @@
 
 __all__ = [
     "DPModel",
-    "DPAtomicModel",
     "PRECISION_DICT",
     "DEFAULT_PRECISION",
     "NativeOP",

diff --git a/deepmd/dpmodel/model/dp_model.py b/deepmd/dpmodel/model/dp_model.py
@@ -1,5 +1,5 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
-from deepmd.dpmodel import (
+from deepmd.dpmodel.atomic_model import (
     DPAtomicModel,
 )
 

diff --git a/source/tests/pt/model/test_dp_atomic_model.py b/source/tests/pt/model/test_dp_atomic_model.py
@@ -4,7 +4,7 @@
 import numpy as np
 import torch
 
-from deepmd.dpmodel import DPAtomicModel as DPDPAtomicModel
+from deepmd.dpmodel.atomic_model import DPAtomicModel as DPDPAtomicModel
 from deepmd.dpmodel.descriptor import DescrptSeA as DPDescrptSeA
 from deepmd.dpmodel.fitting import InvarFitting as DPInvarFitting
 from deepmd.pt.model.atomic_model import (