From 22e1fd50bb6ff90adf9d533d0bfb8bda4b1414d9 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 6 Feb 2025 19:36:10 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- deepmd/tf/fit/polar.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/deepmd/tf/fit/polar.py b/deepmd/tf/fit/polar.py index 020f52bf05..e7dcc57f71 100644 --- a/deepmd/tf/fit/polar.py +++ b/deepmd/tf/fit/polar.py @@ -575,7 +575,9 @@ def init_variables( graph, f"fitting_attr{suffix}/t_bias_atom_polar" ) 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.") + 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." + ) pass def enable_mixed_precision(self, mixed_prec: Optional[dict] = None) -> None: