diff --git a/simba/data_processors/cuda/euclidean_distances_cuda.py b/simba/data_processors/cuda/euclidean_distances_cuda.py index 0245a4741..b4bc909d4 100644 --- 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 index 1ee74f058..78b18b56d 100644 --- 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):