check ut

deepmodeling · Jan 6, 2025 · 3ccb4b9 · 3ccb4b9
1 parent 6a5d169
commit 3ccb4b9
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/source/tests/pt/test_make_stat_input.py b/source/tests/pt/test_make_stat_input.py
@@ -79,7 +79,6 @@ def test_make_stat_input(self):
         )
         bias, _ = compute_output_stats(lst, ntypes=57)
         energy = bias.get("energy")
-        print(energy)
         non_zero_count = self.count_non_zero_elements(energy)
         self.assertEqual(
             non_zero_count,
@@ -88,7 +87,7 @@ def test_make_stat_input(self):
         )
 
     def test_make_stat_input_nocomplete(self):
-        #missing element：13,31,37
+        #missing element:13,31,37
         #only one frame would be count
 
         lst = make_stat_input(
@@ -100,7 +99,6 @@ def test_make_stat_input_nocomplete(self):
         )
         bias, _ = compute_output_stats(lst, ntypes=57)
         energy = bias.get("energy")
-        print(energy)
         non_zero_count = self.count_non_zero_elements(energy)
         self.assertLess(
             non_zero_count,