diff --git a/deepmd/tf/descriptor/se_atten_v2.py b/deepmd/tf/descriptor/se_atten_v2.py
index 6534e8562f..a4fdf24a55 100644
--- a/deepmd/tf/descriptor/se_atten_v2.py
+++ b/deepmd/tf/descriptor/se_atten_v2.py
@@ -8,6 +8,7 @@
 from deepmd.utils.version import (
     check_version_compatibility,
 )
+
 from .descriptor import (
     Descriptor,
 )