add main for test

deepmodeling · Nov 5, 2024 · c91b8bb · c91b8bb
1 parent 14ac57b
commit c91b8bb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/source/tests/pt/test_model_compression_se_a.py b/source/tests/pt/test_model_compression_se_a.py
@@ -13,7 +13,7 @@
 )
 
 # from deepmd.tf.entrypoints.compress import compress
-from source.tests.pt.common import (
+from .common import (
     j_loader,
     run_dp,
     tests_path,
@@ -573,3 +573,7 @@ def test_2frame_atm(self):
         np.testing.assert_almost_equal(av0, av1, default_places)
         np.testing.assert_almost_equal(ee0, ee1, default_places)
         np.testing.assert_almost_equal(vv0, vv1, default_places)
+
+
+if __name__ == "__main__":
+    unittest.main()