Fix import sorting with isort

cupy_similarity.py

@@ -38,9 +38,10 @@
     >>>     print("GPU not available, using CPU fallback")
 """
 
-import numpy as np
-from typing import Union, Optional, Tuple
 import warnings
+from typing import Optional, Tuple, Union
+
+import numpy as np
 
 # Try to import CuPy for GPU acceleration
 try:

examples_gpu.py

@@ -17,13 +17,15 @@
     python examples_gpu.py
 """
 
-import numpy as np
 import time
 from typing import Tuple
 
+import numpy as np
+
+import gpu_similarity as gpu_sim
+
 # Import both CPU and GPU implementations
 from similarity import lorentz_similarity, standard_cosine_similarity
-import gpu_similarity as gpu_sim
 
 
 def print_section(title: str):

examples_loop_prevention.py

@@ -13,6 +13,7 @@
 """
 
 import numpy as np
+
 from similarity import lorentz_similarity, standard_cosine_similarity
 
 

similarity.py

@@ -35,8 +35,9 @@
 preventing self-reinforcing loops.
 """
 
+from typing import Optional, Union
+
 import numpy as np
-from typing import Union, Optional
 
 
 def lorentz_similarity(u: np.ndarray, v: np.ndarray, epsilon: float = 1e-10) -> float:

test_gpu_similarity.py

@@ -12,6 +12,7 @@
 
 import numpy as np
 import pytest
+
 import gpu_similarity as gpu_sim
 from similarity import lorentz_similarity, standard_cosine_similarity
 

test_pytorch_modules.py

@@ -10,13 +10,13 @@
 import torch
 import torch.nn as nn
 
+from eigen_attention import EigenAttention
+from eigen_memory import EigenMemory
 from gpu_similarity import (
+    compare_self_similarity_torch,
     eigen_similarity,
     standard_cosine_similarity_torch,
-    compare_self_similarity_torch,
 )
-from eigen_memory import EigenMemory
-from eigen_attention import EigenAttention
 
 
 class TestGPUSimilarity:
@@ -422,6 +422,7 @@ def test_consistency_with_numpy_version(self):
         """Test that PyTorch version is consistent with NumPy version."""
         # Import NumPy version
         import numpy as np
+
         from similarity import lorentz_similarity
 
         # Create same vector in both frameworks

test_similarity.py

@@ -8,7 +8,8 @@
 
 import numpy as np
 import pytest
-from similarity import lorentz_similarity, standard_cosine_similarity, compare_self_similarity
+
+from similarity import compare_self_similarity, lorentz_similarity, standard_cosine_similarity
 
 
 class TestSelfReferenceProperty:

tests/test_eigen_modules.py

@@ -1,8 +1,9 @@
 import unittest
+
 import torch
 
-from eigen_memory import EigenMemory
 from eigen_attention import EigenAttention
+from eigen_memory import EigenMemory
 from eigen_transformer import EigenTransformerBlock
 from gpu_similarity import eigen_similarity