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

pre-commit-ci[bot] · pre-commit-ci[bot] · commit 36d46778f7d6 · 2025-02-06T20:15:12.000Z
for more information, see https://pre-commit.ci
diff --git a/deepmd/tf/fit/polar.py b/deepmd/tf/fit/polar.py
@@ -577,7 +577,7 @@ def init_variables(
             except GraphWithoutTensorError:
                 warnings.warn(
                     "You are trying to read a model trained with shift_diag=True, but the mean of the diagonal terms of the polarizability is not stored in the graph. This will lead to wrong inference results. You may train your model with the latest DeePMD-kit to avoid this issue.",
-                    stacklevel=2
+                    stacklevel=2,
                 )
 
     def enable_mixed_precision(self, mixed_prec: Optional[dict] = None) -> None:

Original file line number	Diff line number	Diff line change
`@@ -577,7 +577,7 @@ def init_variables(`
`577`	`577`	`except GraphWithoutTensorError:`
`578`	`578`	`warnings.warn(`
`579`	`579`	`"You are trying to read a model trained with shift_diag=True, but the mean of the diagonal terms of the polarizability is not stored in the graph. This will lead to wrong inference results. You may train your model with the latest DeePMD-kit to avoid this issue.",`
`580`		`- stacklevel=2`
	`580`	`+ stacklevel=2,`
`581`	`581`	`)`
`582`	`582`
`583`	`583`	`def enable_mixed_precision(self, mixed_prec: Optional[dict] = None) -> None:`