From 603128add336511b961ebcf6882a3213cb25e375 Mon Sep 17 00:00:00 2001
From: Duo <50307526+iProzd@users.noreply.github.com>
Date: Thu, 1 Feb 2024 23:05:52 +0800
Subject: [PATCH] fix uts

---
 source/tests/pt/models/dpa2_tebd.pth    | Bin 0 -> 1085 bytes
 source/tests/pt/test_descriptor_dpa2.py |   2 +-
 2 files changed, 1 insertion(+), 1 deletion(-)
 create mode 100644 source/tests/pt/models/dpa2_tebd.pth

diff --git a/source/tests/pt/models/dpa2_tebd.pth b/source/tests/pt/models/dpa2_tebd.pth
new file mode 100644
index 0000000000000000000000000000000000000000..3d4fc5511c93036a18be1b290fcdecc482215bf6
GIT binary patch
literal 1085
zcmWIWW@cev;NW1u0AdV047vF!sX6iGshQ~+CB^zFi6x181=%@nP8<wTP|(1XoSdJN
zlbT$TnV(n8<zJMNT9lgNl9^n>Rmh+jRLB@105mx@Hz_qGB{MHw4`Nm!Q*uduQF4Y}
zd}&E$PBB+}QEF0YW==|cNornkeo=gx5mzCzhDHQCP;W6%Wny}2AqzwcmrH(WQch|x
zM3k$LHG&bS$e@rdf*D9B7nByVdrOoQas+xacr$x*v=wrCb9ghh6>@cEfZW4fQpgij
z$Xi@n$QQv0)DWMWT9OFzSRp@HUO*!PD9TVMSX(Frw!SR2s2J$ALSb*lVz3UlUkXKP
z3q><P;-$q&dI8?-9Q|j{W^)3a1Hu6~16G8=6idMB8@M?c!h;nSyaega7|BC|wIA%a
z9@}-3H*nE@KT+*Vx8{GezwEeJ#KQU7ejUZYtzQ-&+W#(Y+ok$|_xro@!>73%-(Y`P
z?Y_N0-0}TyYI%RGUVnUlUGB=6m&4xKzj|P0d}!wz`|9Kr2MOK>`?tipFM7!OWB;uA
z(`2T<c(DKWp{f3ovp3kAJX$fUO!(b?P+(C4K*5%{A=u~u&=3&D9c-cuwrIfyjT$#6
zOVYy)7O=)#AQxPmmmbOlv>k*4yct0hJWn7;su)ND1z^u1=w=}MPy|JPBrvI9nxSui
zZUV9kMNv$c0CX*?3D9r}@MdGvfhv__)`e>V<r5Hq(I+8X1|g^bP#-AYG2DjoLHR4d
Po0SbD#teiY^$@iHqv;JY

literal 0
HcmV?d00001

diff --git a/source/tests/pt/test_descriptor_dpa2.py b/source/tests/pt/test_descriptor_dpa2.py
index 093f53bc32..22371f612d 100644
--- 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.pt"
+        self.file_type_embed = Path(CUR_DIR) / "models" / "dpa2_tebd.pth"
 
     def test_descriptor_hyb(self):
         # torch.manual_seed(0)