fix uts

deepmodeling · Feb 1, 2024 · 42c75de · 42c75de
1 parent 8fec4cb
commit 42c75de
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/source/tests/pt/test_descriptor_dpa2.py b/source/tests/pt/test_descriptor_dpa2.py
@@ -112,7 +112,7 @@ def setUp(self):
         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.pt"
-        self.file_type_embed = Path(CUR_DIR) / "models" / "dpa2_tebd.pth"
+        self.file_type_embed = Path(CUR_DIR) / "models" / "dpa2_tebd.pt"
 
     def test_descriptor_hyb(self):
         # torch.manual_seed(0)

diff --git a/source/tests/pt/test_fitting_net.py b/source/tests/pt/test_fitting_net.py
@@ -111,7 +111,7 @@ def test_consistency(self):
         ).to(env.DEVICE)
         for name, param in my_fn.named_parameters():
             matched = re.match(
-                "filter_layers\.networks\.(\d).layers\.(\d)\.([a-z]+)", name
+                "filter_layers\._networks\.(\d).layers\.(\d)\.([a-z]+)", name
             )
             key = None
             if matched: