use stacklevel=2 for warning

Yi-FanLi · Yi-FanLi · commit e8bdc392b662 · 2025-02-06T15:13:21.000-05:00
diff --git a/deepmd/tf/fit/polar.py b/deepmd/tf/fit/polar.py
@@ -576,7 +576,8 @@ 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."
+                    "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
                 )
 
     def enable_mixed_precision(self, mixed_prec: Optional[dict] = None) -> None:

Original file line number	Diff line number	Diff line change
`@@ -576,7 +576,8 @@ def init_variables(`
`576`	`576`	`)`
`577`	`577`	`except GraphWithoutTensorError:`
`578`	`578`	`warnings.warn(`
`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."`
	`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`	`581`	`)`
`581`	`582`
`582`	`583`	`def enable_mixed_precision(self, mixed_prec: Optional[dict] = None) -> None:`