style: fix import order with isort

benchmarks.py

@@ -7,10 +7,11 @@
 
 from __future__ import annotations
 
+import time
+from typing import Callable
+
 import torch
 import torch.nn as nn
-from typing import Callable
-import time
 
 
 class BenchmarkTask:
@@ -304,7 +305,7 @@ def print_results(self, results: list[dict]):
 
 def run_benchmarks():
     """Run benchmarks comparing standard vs spacetime models."""
-    from demo_loop_prevention import StandardReasoningModel, SpacetimeReasoningModel
+    from demo_loop_prevention import SpacetimeReasoningModel, StandardReasoningModel
 
     print("=" * 80)
     print("Loop-Prone Reasoning Benchmarks")

chain_validator.py

@@ -16,9 +16,10 @@
 
 from __future__ import annotations
 
+from typing import List, Tuple
+
 import torch
 import torch.nn as nn
-from typing import List, Tuple
 
 
 def create_reasoning_chain(chain_type: str, length: int = 5, dim: int = 64) -> torch.Tensor:

demo_loop_prevention.py

@@ -12,12 +12,13 @@
 
 from __future__ import annotations
 
+from typing import Optional
+
 import torch
 import torch.nn as nn
-from typing import Optional
 
-from standard_attention import StandardAttention
 from spacetime_feedback import SpacetimeFeedbackBlock
+from standard_attention import StandardAttention
 
 
 class StandardReasoningModel(nn.Module):

paradox_detector.py

@@ -15,10 +15,11 @@
 
 from __future__ import annotations
 
+from enum import Enum
+from typing import List, Tuple
+
 import torch
 import torch.nn as nn
-from typing import Tuple, List
-from enum import Enum
 
 
 class LogicType(Enum):