more linting

Signed-off-by: Antoni Viros i Martin <[email protected]>

Author: ani300

fms_mo/aiu_addons/fp8/fp8_attn.py

@@ -18,7 +18,6 @@
 import math
 
 # Third Party
-from fms.modules.attention import _sdpa_compute_op
 import torch
 
 # Local
@@ -30,6 +29,7 @@
     # Third Party
     from fms.modules.attention import (
         AttentionKwargs,
+        _sdpa_compute_op,
         _sdpa_update_attn_kwargs,
         register_attention_op,
     )