diff --git a/simba/mixins/feature_extraction_mixin.py b/simba/mixins/feature_extraction_mixin.py index c3ad33aa7..223f80220 100644 --- a/simba/mixins/feature_extraction_mixin.py +++ b/simba/mixins/feature_extraction_mixin.py @@ -19,7 +19,7 @@ import simba from simba.utils.checks import (check_file_exist_and_readable, check_if_filepath_list_is_empty, - check_minimum_roll_windows, check) + check_minimum_roll_windows) from simba.utils.enums import Options, Paths from simba.utils.errors import CountError from simba.utils.read_write import (get_bp_headers, read_config_file, diff --git a/simba/utils/checks.py b/simba/utils/checks.py index c3720c355..8cef2d25f 100644 --- a/simba/utils/checks.py +++ b/simba/utils/checks.py @@ -804,7 +804,7 @@ def check_valid_array(data: np.ndarray, accepted_sizes: Optional[List[int]] = None, accepted_axis_0_shape: Optional[List[int]] = None, accepted_axis_1_shape: Optional[List[int]] = None, - accepted_dtypes: Optional[Union[List[Union[str, Type]], Tuple[Union[str, Type]]], Iterable[Any]] = None, + accepted_dtypes: Optional[Union[List[Union[str, Type]], Tuple[Union[str, Type]], Iterable[Any]]] = None, accepted_values: Optional[List[Any]] = None, accepted_shapes: Optional[List[Tuple[int]]] = None, min_axis_0: Optional[int] = None,