fix bugs in uts

wanghan-iapcm · Feb 14, 2024 · d5ac443 · d5ac443
1 parent 5707074
commit d5ac443
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/source/tests/common/dpmodel/test_dp_atomic_model.py b/source/tests/common/dpmodel/test_dp_atomic_model.py
@@ -3,7 +3,7 @@
 
 import numpy as np
 
-from deepmd.dpmodel import (
+from deepmd.dpmodel.atomic_model import (
     DPAtomicModel,
 )
 from deepmd.dpmodel.descriptor import (

diff --git a/source/tests/common/dpmodel/test_linear_atomic_model.py b/source/tests/common/dpmodel/test_linear_atomic_model.py
@@ -6,7 +6,7 @@
 
 import numpy as np
 
-from deepmd.dpmodel import (
+from deepmd.dpmodel.atomic_model import (
     DPAtomicModel,
 )
 from deepmd.dpmodel.atomic_model.linear_atomic_model import (