From 49535b98483b513c2208e4d831c2c47ee8911ea0 Mon Sep 17 00:00:00 2001 From: Claude Date: Fri, 14 Nov 2025 05:04:15 +0000 Subject: [PATCH] Fix import sorting with isort Run isort to fix import ordering in Python files to comply with the Code Quality lint checks in CI. This ensures consistent import formatting across the codebase. --- cupy_similarity.py | 5 +++-- examples_gpu.py | 6 ++++-- examples_loop_prevention.py | 1 + similarity.py | 3 ++- test_gpu_similarity.py | 1 + test_pytorch_modules.py | 7 ++++--- test_similarity.py | 3 ++- tests/test_eigen_modules.py | 3 ++- 8 files changed, 19 insertions(+), 10 deletions(-) diff --git a/cupy_similarity.py b/cupy_similarity.py index 5008197..3e164f7 100644 --- a/cupy_similarity.py +++ b/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: diff --git a/examples_gpu.py b/examples_gpu.py index 2a7075e..185cf50 100644 --- a/examples_gpu.py +++ b/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): diff --git a/examples_loop_prevention.py b/examples_loop_prevention.py index dc7fdbf..27ac9ed 100644 --- a/examples_loop_prevention.py +++ b/examples_loop_prevention.py @@ -13,6 +13,7 @@ """ import numpy as np + from similarity import lorentz_similarity, standard_cosine_similarity diff --git a/similarity.py b/similarity.py index bddd1ad..6335968 100644 --- a/similarity.py +++ b/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: diff --git a/test_gpu_similarity.py b/test_gpu_similarity.py index fc9765d..18379f3 100644 --- a/test_gpu_similarity.py +++ b/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 diff --git a/test_pytorch_modules.py b/test_pytorch_modules.py index 51e1a66..8629b0d 100644 --- a/test_pytorch_modules.py +++ b/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 diff --git a/test_similarity.py b/test_similarity.py index ead34fa..facef2e 100644 --- a/test_similarity.py +++ b/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: diff --git a/tests/test_eigen_modules.py b/tests/test_eigen_modules.py index d30373e..47f70c4 100644 --- a/tests/test_eigen_modules.py +++ b/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