fix: precommit

deepmodeling · Mar 7, 2024 · 254c198 · 254c198
1 parent 7748723
commit 254c198
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/source/tests/pt/model/test_polar_stat.py b/source/tests/pt/model/test_polar_stat.py
@@ -41,7 +41,7 @@ def setUp(self) -> None:
             ntypes=ntypes,
             dim_descrpt=1,
             embedding_width=1,
-            sel_type=[i for i in range(ntypes)],
+            sel_type=list(range(ntypes)),
         )
         self.ptpolar = PolarFittingNet(
             ntypes=ntypes,
@@ -54,7 +54,6 @@ def test_atomic_consistency(self):
         tfbias = self.tfpolar.constant_matrix
         self.ptpolar.compute_output_stats(self.sampled)
         ptbias = self.ptpolar.constant_matrix
-        print(tfbias, to_numpy_array(ptbias))
         np.testing.assert_allclose(tfbias, to_numpy_array(ptbias))
 
     def test_global_consistency(self):
@@ -70,5 +69,4 @@ def test_global_consistency(self):
         tfbias = self.tfpolar.constant_matrix
         self.ptpolar.compute_output_stats(self.sampled)
         ptbias = self.ptpolar.constant_matrix
-        print(tfbias, to_numpy_array(ptbias))
         np.testing.assert_allclose(tfbias, to_numpy_array(ptbias))