diff --git a/simba/model/grid_search_rf.py b/simba/model/grid_search_rf.py index 35ddb50ac..e41ebc5ec 100644 --- a/simba/model/grid_search_rf.py +++ b/simba/model/grid_search_rf.py @@ -9,10 +9,12 @@ from simba.mixins.config_reader import ConfigReader from simba.mixins.train_model_mixin import TrainModelMixin from simba.utils.checks import check_if_filepath_list_is_empty -from simba.utils.enums import (ConfigKey, Dtypes, Formats, Methods, MLParamKeys, Options) +from simba.utils.enums import (ConfigKey, Dtypes, Formats, Methods, + MLParamKeys, Options) from simba.utils.errors import InvalidInputError, NoDataError from simba.utils.printing import stdout_success -from simba.utils.read_write import (read_config_entry, read_simba_meta_files, write_df) +from simba.utils.read_write import (read_config_entry, read_simba_meta_files, + write_df) class GridSearchRandomForestClassifier(ConfigReader, TrainModelMixin): diff --git a/simba/model/inference_batch.py b/simba/model/inference_batch.py index 4df1256fe..042b59c18 100644 --- a/simba/model/inference_batch.py +++ b/simba/model/inference_batch.py @@ -2,18 +2,21 @@ import os from copy import deepcopy -from typing import Union, Optional +from typing import Optional, Union import numpy as np from simba.mixins.config_reader import ConfigReader from simba.mixins.train_model_mixin import TrainModelMixin -from simba.utils.checks import (check_all_file_names_are_represented_in_video_log, check_if_keys_exist_in_dict, check_if_dir_exists) +from simba.utils.checks import ( + check_all_file_names_are_represented_in_video_log, check_if_dir_exists, + check_if_keys_exist_in_dict) from simba.utils.data import plug_holes_shortest_bout from simba.utils.enums import TagNames from simba.utils.errors import NoFilesFoundError from simba.utils.printing import SimbaTimer, log_event, stdout_success -from simba.utils.read_write import get_fn_ext, read_df, write_df, find_files_of_filetypes_in_directory +from simba.utils.read_write import (find_files_of_filetypes_in_directory, + get_fn_ext, read_df, write_df) class InferenceBatch(TrainModelMixin, ConfigReader): diff --git a/simba/model/train_rf.py b/simba/model/train_rf.py index 5e8a35adc..25ff3eeb2 100644 --- a/simba/model/train_rf.py +++ b/simba/model/train_rf.py @@ -2,15 +2,17 @@ import ast import os -from typing import Union, Optional +from typing import Optional, Union import pandas as pd from sklearn.model_selection import train_test_split from simba.mixins.config_reader import ConfigReader from simba.mixins.train_model_mixin import TrainModelMixin -from simba.utils.checks import check_if_filepath_list_is_empty, check_int, check_if_dir_exists -from simba.utils.enums import (ConfigKey, Dtypes, Formats, Methods, MLParamKeys, Options) +from simba.utils.checks import (check_if_dir_exists, + check_if_filepath_list_is_empty, check_int) +from simba.utils.enums import (ConfigKey, Dtypes, Formats, Methods, + MLParamKeys, Options) from simba.utils.printing import SimbaTimer, stdout_success from simba.utils.read_write import read_config_entry, write_df diff --git a/simba/video_processors/video_processing.py b/simba/video_processors/video_processing.py index 9816a065f..6d7f570d4 100644 --- a/simba/video_processors/video_processing.py +++ b/simba/video_processors/video_processing.py @@ -46,8 +46,9 @@ NoFilesFoundError, NotDirectoryError, ResolutionError) from simba.utils.lookups import (get_ffmpeg_crossfade_methods, get_fonts, - percent_to_crf_lookup, percent_to_qv_lk, - video_quality_to_preset_lookup, get_named_colors) + get_named_colors, percent_to_crf_lookup, + percent_to_qv_lk, + video_quality_to_preset_lookup) from simba.utils.printing import SimbaTimer, stdout_success from simba.utils.read_write import ( check_if_hhmmss_timestamp_is_valid_part_of_video,