Add 2 UT

deepmodeling · May 24, 2024 · 68eda81 · 68eda81
1 parent d8dadda
commit 68eda81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/tests/pt/test_dp_show.py b/source/tests/pt/test_dp_show.py
@@ -58,7 +58,6 @@ def test_checkpoint(self):
             in results[-1]
         )
 
-    '''
     def test_frozen_model(self):
         INPUT = "frozen_model.pth"
         ATTRIBUTES = "type-map descriptor fitting-net"
@@ -74,6 +73,7 @@ def test_frozen_model(self):
             in results[-1]
         )
 
+    '''
     def test_checkpoint_error(self):
         INPUT = "model.pt"
         ATTRIBUTES = "model-branch type-map descriptor fitting-net"