From e4a6d05ec64cf89ec6f2f91903f601d736c35a79 Mon Sep 17 00:00:00 2001 From: sronilsson Date: Tue, 10 Sep 2024 20:58:46 +0000 Subject: [PATCH] cleaned --- simba/data_processors/cuda/timeseries.py | 11 ++++++----- simba/data_processors/cuda/utils.py | 4 +++- .../feature_extractor_user_defined.py | 2 +- 3 files changed, 10 insertions(+), 7 deletions(-) diff --git a/simba/data_processors/cuda/timeseries.py b/simba/data_processors/cuda/timeseries.py index f80ede43f..f34c443df 100644 --- a/simba/data_processors/cuda/timeseries.py +++ b/simba/data_processors/cuda/timeseries.py @@ -1,11 +1,12 @@ -import numpy as np -from typing import Optional from time import perf_counter -from simba.utils.enums import Formats -from simba.utils.checks import check_valid_array, check_float +from typing import Optional + +import numpy as np from numba import cuda -from simba.data_processors.cuda.utils import _cuda_mean, _cuda_std +from simba.data_processors.cuda.utils import _cuda_mean, _cuda_std +from simba.utils.checks import check_float, check_valid_array +from simba.utils.enums import Formats THREADS_PER_BLOCK = 1024 diff --git a/simba/data_processors/cuda/utils.py b/simba/data_processors/cuda/utils.py index 5d7f89eb8..cfb9875e5 100644 --- a/simba/data_processors/cuda/utils.py +++ b/simba/data_processors/cuda/utils.py @@ -1,6 +1,8 @@ +import math + import numpy as np from numba import cuda -import math + @cuda.jit(device=True) def _cuda_sum(x: np.ndarray): diff --git a/simba/feature_extractors/feature_extractor_user_defined.py b/simba/feature_extractors/feature_extractor_user_defined.py index db161d067..8c5590e29 100644 --- a/simba/feature_extractors/feature_extractor_user_defined.py +++ b/simba/feature_extractors/feature_extractor_user_defined.py @@ -10,7 +10,7 @@ from simba.mixins.config_reader import ConfigReader from simba.mixins.feature_extraction_mixin import FeatureExtractionMixin -from simba.utils.checks import check_str, check_float +from simba.utils.checks import check_float, check_str from simba.utils.errors import MissingColumnsError, ParametersFileError from simba.utils.printing import SimbaTimer, stdout_success from simba.utils.read_write import get_fn_ext, read_df, write_df