commit

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
deepmodeling · Oct 3, 2024 · ecc9e7d · ecc9e7d
1 parent f6190e3
commit ecc9e7d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -6,7 +6,8 @@
 import pytest
 from dargs.check import check
 from deepmd.utils.argcheck import gen_args
-from deepmd_mace.argcheck import mace_model_args  # noqa: F401
+
+from deepmd_gnn.argcheck import mace_model_args  # noqa: F401
 
 example_path = Path(__file__).parent.parent / "examples"
 

diff --git a/tests/test_model.py b/tests/test_model.py
@@ -23,8 +23,9 @@
     to_numpy_array,
     to_torch_tensor,
 )
-from deepmd_mace.mace import MaceModel
-from deepmd_mace.nequip import NequipModel
+
+from deepmd_gnn.mace import MaceModel
+from deepmd_gnn.nequip import NequipModel
 
 GLOBAL_SEED = 20240822