Automatic pre-commit fixes

Author: aadya940

aeon/transformations/collection/dictionary_based/_paa.py

@@ -3,7 +3,7 @@
 __maintainer__ = []
 
 import numpy as np
-from numba import njit, prange, get_num_threads, set_num_threads
+from numba import get_num_threads, njit, prange, set_num_threads
 
 from aeon.transformations.collection import BaseCollectionTransformer
 from aeon.utils.validation import check_n_jobs

test.py

@@ -8,11 +8,11 @@
 #     """
 #     x_flat = x.flatten()
 #     result = np.empty(x_flat.shape[0], dtype=np.intp)
-    
+
 #     for i in range(x_flat.shape[0]):
 #         val = x_flat[i]
 #         bin_idx = 0
-        
+
 #         if right:
 #             # bins[i] < x <= bins[i+1]
 #             for j in range(len(bins)):
@@ -27,23 +27,24 @@
 #                     bin_idx = j
 #                     break
 #                 bin_idx = j + 1
-        
+
 #         result[i] = bin_idx
-    
+
 #     return result.reshape(x.shape)
 
+
 @njit(fastmath=True, cache=True, parallel=True)
 def numba_digitize_parallel(x, bins, right=False):
     """
     Parallel version for better performance on large arrays.
     """
     x_flat = x.flatten()
     result = np.empty(x_flat.shape[0], dtype=np.intp)
-    
+
     for i in prange(x_flat.shape[0]):
         val = x_flat[i]
         bin_idx = 0
-        
+
         if right:
             for j in range(len(bins)):
                 if val <= bins[j]:
@@ -56,9 +57,9 @@ def numba_digitize_parallel(x, bins, right=False):
                     bin_idx = j
                     break
                 bin_idx = j + 1
-        
+
         result[i] = bin_idx
-    
+
     return result.reshape(x.shape)
 
 
@@ -83,16 +84,25 @@ def _parallel_get_sax_symbols(X, breakpoints):
 if __name__ == "__main__":
     x = np.array([[[0.2, 6.4, 3.0, 1.6]]])
     bins = np.array([0.0, 1.0, 2.5, 4.0, 10.0])
-    
+
     print("Original:", np.digitize(x, bins))
     print("Numba:   ", numba_digitize_parallel(x, bins))
-    print("Match:   ", np.array_equal(np.digitize(x, bins), numba_digitize_parallel(x, bins)))
+    print(
+        "Match:   ",
+        np.array_equal(np.digitize(x, bins), numba_digitize_parallel(x, bins)),
+    )
 
     print("Curr: ", _parallel_get_sax_symbols(x, bins))
-    
+
     # Test with right=True
     print("\nWith right=True:")
     print("Original:", np.digitize(x, bins, right=True))
     print("Numba:   ", numba_digitize_parallel(x, bins, right=True))
-    print("Match:   ", np.array_equal(np.digitize(x, bins, right=True), numba_digitize_parallel(x, bins, right=True)))
-    print("Curr: ", _parallel_get_sax_symbols(x, bins))
+    print(
+        "Match:   ",
+        np.array_equal(
+            np.digitize(x, bins, right=True),
+            numba_digitize_parallel(x, bins, right=True),
+        ),
+    )
+    print("Curr: ", _parallel_get_sax_symbols(x, bins))