diff --git a/simba/bounding_box_tools/yolo/geometries_to_annotations.py b/simba/bounding_box_tools/yolo/geometries_to_annotations.py index 9d8a792f4..1134d867f 100644 --- a/simba/bounding_box_tools/yolo/geometries_to_annotations.py +++ b/simba/bounding_box_tools/yolo/geometries_to_annotations.py @@ -10,9 +10,12 @@ from skimage.draw import polygon from simba.mixins.geometry_mixin import GeometryMixin -from simba.utils.checks import check_instance, check_int, check_valid_array, check_if_dir_exists +from simba.utils.checks import (check_if_dir_exists, check_instance, check_int, + check_valid_array) from simba.utils.enums import Formats -from simba.utils.read_write import (get_video_meta_data, read_df, read_frm_of_video, find_files_of_filetypes_in_directory) +from simba.utils.read_write import (find_files_of_filetypes_in_directory, + get_video_meta_data, read_df, + read_frm_of_video) def geometry_to_rle(geometry: Union[np.ndarray, Polygon], img_size: Tuple[int, int]): diff --git a/simba/data_processors/boolean_conditional_calculator.py b/simba/data_processors/boolean_conditional_calculator.py index 01c8b52b1..45962f5aa 100644 --- a/simba/data_processors/boolean_conditional_calculator.py +++ b/simba/data_processors/boolean_conditional_calculator.py @@ -6,11 +6,14 @@ import pandas as pd from simba.mixins.config_reader import ConfigReader -from simba.utils.checks import (check_all_file_names_are_represented_in_video_log, check_if_df_field_is_boolean, check_instance) +from simba.utils.checks import ( + check_all_file_names_are_represented_in_video_log, + check_if_df_field_is_boolean, check_instance) +from simba.utils.data import detect_bouts from simba.utils.errors import MissingColumnsError from simba.utils.printing import stdout_success -from simba.utils.read_write import (get_fn_ext, read_data_paths, read_df, read_video_info, str_2_bool) -from simba.utils.data import detect_bouts +from simba.utils.read_write import (get_fn_ext, read_data_paths, read_df, + read_video_info, str_2_bool) class BooleanConditionalCalculator(ConfigReader): diff --git a/simba/mixins/train_model_mixin.py b/simba/mixins/train_model_mixin.py index 60ae511c9..56b1c9e54 100644 --- a/simba/mixins/train_model_mixin.py +++ b/simba/mixins/train_model_mixin.py @@ -32,7 +32,8 @@ from sklearn.inspection import partial_dependence, permutation_importance from sklearn.metrics import classification_report, precision_recall_curve from sklearn.model_selection import ShuffleSplit, learning_curve -from sklearn.preprocessing import (MinMaxScaler, QuantileTransformer, StandardScaler) +from sklearn.preprocessing import (MinMaxScaler, QuantileTransformer, + StandardScaler) from sklearn.tree import export_graphviz from sklearn.utils import parallel_backend from tabulate import tabulate @@ -57,11 +58,15 @@ from simba.plotting.shap_agg_stats_visualizer import \ ShapAggregateStatisticsVisualizer from simba.ui.tkinter_functions import TwoOptionQuestionPopUp -from simba.utils.checks import (check_file_exist_and_readable, check_float, check_if_dir_exists, check_if_valid_input, check_instance, check_int, check_str, check_that_column_exist, check_valid_dataframe, check_valid_lst) +from simba.utils.checks import (check_file_exist_and_readable, check_float, + check_if_dir_exists, check_if_valid_input, + check_instance, check_int, check_str, + check_that_column_exist, check_valid_dataframe, + check_valid_lst) from simba.utils.data import (detect_bouts, detect_bouts_multiclass, get_library_version) -from simba.utils.enums import (OS, ConfigKey, Defaults, Dtypes, Methods, - MLParamKeys, Options, Formats) +from simba.utils.enums import (OS, ConfigKey, Defaults, Dtypes, Formats, + Methods, MLParamKeys, Options) from simba.utils.errors import (ClassifierInferenceError, ColumnNotFoundError, CorruptedFileError, DataHeaderError, FaultyTrainingSetError,