Merge remote-tracking branch 'origin/master'

sgoldenlab · Aug 30, 2024 · b8cdf6a · b8cdf6a
2 parents 4f1d272 + 4f32016
commit b8cdf6a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/simba/data_processors/cuda/euclidean_distances_cuda.py b/simba/data_processors/cuda/euclidean_distances_cuda.py
@@ -1,8 +1,10 @@
+from typing import Optional
+
+import cupy as cp
 import numpy as np
-from simba.utils.checks import check_valid_array, check_int
+
+from simba.utils.checks import check_int, check_valid_array
 from simba.utils.enums import Formats
-import cupy as cp
-from typing import Optional
 
 
 def get_euclidean_distance_cupy(x: np.ndarray,

diff --git a/simba/data_processors/cuda/img_stack_mse.py b/simba/data_processors/cuda/img_stack_mse.py
@@ -1,7 +1,11 @@
+from typing import Optional
+
 import numpy as np
 from numba import cuda
-from simba.utils.checks import check_if_valid_img, check_instance, check_valid_array
-from typing import Optional
+
+from simba.utils.checks import (check_if_valid_img, check_instance,
+                                check_valid_array)
+
 
 @cuda.jit()
 def _grey_mse(data, ref_img, stride, batch_cnt, mse_arr):