fix uts

dptech-corp · Jan 24, 2024 · 941ec7b · 941ec7b
1 parent 8b6c1bd
commit 941ec7b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/tests/models/dpa2.pth → tests/models/dpa2.pt b/tests/models/dpa2.pth → tests/models/dpa2.pt
diff --git a/tests/test_descriptor_dpa2.py b/tests/test_descriptor_dpa2.py
@@ -37,7 +37,7 @@ def setUp(self):
     ]).to(env.DEVICE)
     with open(Path(CUR_DIR)/"models"/"dpa2_hyb.json") as fp:
       self.model_json = json.load(fp)
-    self.file_model_param = Path(CUR_DIR)/"models"/"dpa2.pth"
+    self.file_model_param = Path(CUR_DIR)/"models"/"dpa2.pt"
     self.file_type_embed = Path(CUR_DIR)/"models"/"dpa2_tebd.pth"
 
   def test_descriptor_hyb(self):