fix model path

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Feb 28, 2024 · f4a3271 · f4a3271
1 parent 99dd474
commit f4a3271
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/source/ipi/tests/test_driver.py b/source/ipi/tests/test_driver.py
@@ -254,10 +254,9 @@ def test_normalize_coords(self):
 
 
 class TestDPIPIPt(TestDPIPI):
-    # copy from test_deeppot_a.py
     @classmethod
     def setUpClass(cls):
-        cls.model_file = "deeppot_sea.pth"
+        cls.model_file = str(tests_path / "infer" / "deeppot_sea.pth")
 
     def setUp(self):
         super().setUp()