[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

deepmd/pt/model/task/polarizability.py

@@ -27,8 +27,8 @@
     to_numpy_array,
 )
 from deepmd.utils.out_stat import (
-    compute_stats_from_redu,
     compute_stats_from_atomic,
+    compute_stats_from_redu,
 )
 from deepmd.utils.path import (
     DPPath,
@@ -208,8 +208,7 @@ def compute_output_stats(
 
                     if sampled[sys]["find_atomic_polarizability"] > 0.0:
                         sys_atom_polar = compute_stats_from_atomic(
-                            sampled[sys]["atomic_polarizability"],
-                            sampled[sys]["type"]
+                            sampled[sys]["atomic_polarizability"], sampled[sys]["type"]
                         )[0]
                     else:
                         if not sampled[sys]["find_polarizability"] > 0.0:
@@ -218,7 +217,7 @@ def compute_output_stats(
                         for itype in range(self.ntypes):
                             type_mask = sampled[sys]["type"] == itype
                             sys_type_count[:, itype] = type_mask.sum(dim=1)
-                        
+
                         sys_bias_redu = sampled[sys]["polarizability"]
 
                         sys_atom_polar = compute_stats_from_redu(

deepmd/tf/fit/polar.py

@@ -202,7 +202,7 @@ def compute_output_stats(self, all_stat):
                         polar_bias.append(
                             np.sum(
                                 all_stat["atomic_polarizability"][ss][:, index_lis, :],
-                                axis=(0,1),
+                                axis=(0, 1),
                             ).reshape((1, 9))
                         )
                 else:  # No atomic polar in this system, so it should have global polar