fix ruff

Author: ZhangLirong-amd

atom/model_ops/fused_moe/mori_prepare_finalize.py

@@ -3,7 +3,7 @@
 
 import logging
 from functools import lru_cache
-from typing import Any, Callable, Optional
+from typing import Any, Callable
 
 import torch
 

atom/utils/dbo/ubatch_wrapper.py

@@ -18,7 +18,7 @@
 )
 
 from .ubatch_splitting import UBatchSlice
-from .ubatching import TBOContext, make_tbo_contexts
+from .ubatching import make_tbo_contexts
 
 logger = logging.getLogger("atom")
 

atom/utils/forward_context.py

@@ -1,6 +1,7 @@
 # SPDX-License-Identifier: MIT
 # Copyright (C) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
 
+import threading
 from contextlib import contextmanager
 from dataclasses import dataclass, field, fields
 from typing import TYPE_CHECKING, Any, Dict, Optional, Set, Union
@@ -338,7 +339,6 @@ def __post_init__(self):
 _forward_kv_cache_context: Optional[ForwardContext] = ForwardContext()
 
 # Thread-local storage for TBO dual-thread execution
-import threading
 
 _forward_context_local = threading.local()