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

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

Author: pre-commit-ci[bot]

deepmd/pt/train/wrapper.py

@@ -5,9 +5,12 @@
     Union,
 )
 
-from deepmd.pt.utils.env import BF16_AUTOCAST
 import torch
 
+from deepmd.pt.utils.env import (
+    BF16_AUTOCAST,
+)
+
 if torch.__version__.startswith("2"):
     import torch._dynamo