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