cleaned

simba/mixins/train_model_mixin.py

@@ -61,8 +61,8 @@
 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, check_valid_array)
+                                check_that_column_exist, check_valid_array,
+                                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, Formats,

simba/mixins/unsupervised_mixin.py

@@ -1,23 +1,28 @@
 __author__ = "Simon Nilsson"
+import datetime
 from typing import Union
-import pandas as pd
+
 import numpy as np
-import datetime
+import pandas as pd
+
 from simba.data_processors.cuda.utils import _cuda_available
-from simba.utils.checks import check_float, check_int, check_valid_boolean, check_instance, check_valid_array
-from simba.utils.errors import SimBAGPUError
+from simba.utils.checks import (check_float, check_instance, check_int,
+                                check_valid_array, check_valid_boolean)
 from simba.utils.data import get_library_version
-from simba.utils.lookups import get_model_names
 from simba.utils.enums import Formats
+from simba.utils.errors import SimBAGPUError
+from simba.utils.lookups import get_model_names
+
 try:
-    import cuml.umap as cuml_umap
     import cuml.cluster.hdbscan as cuml_hdbscan
+    import cuml.umap as cuml_umap
 except ModuleNotFoundError:
-    import umap as cuml_umap
     import hdbscan as cuml_hdbscan
+    import umap as cuml_umap
     pass
-import umap
 import hdbscan
+import umap
+
 
 class UMLMixin(object):
     def __init__(self):

simba/plotting/heat_mapper_clf.py

@@ -2,16 +2,22 @@
 
 import os
 from typing import List, Union
+
 import cv2
 import numpy as np
+
 from simba.mixins.config_reader import ConfigReader
 from simba.mixins.geometry_mixin import GeometryMixin
 from simba.mixins.plotting_mixin import PlottingMixin
+from simba.utils.checks import (
+    check_all_file_names_are_represented_in_video_log,
+    check_filepaths_in_iterable_exist, check_int, check_str,
+    check_valid_dataframe, check_valid_dict)
 from simba.utils.enums import Formats, Options
-from simba.utils.errors import NoSpecifiedOutputError, InvalidInputError
+from simba.utils.errors import InvalidInputError, NoSpecifiedOutputError
 from simba.utils.printing import SimbaTimer, stdout_success
 from simba.utils.read_write import get_fn_ext, read_df
-from simba.utils.checks import check_all_file_names_are_represented_in_video_log, check_str, check_valid_dict, check_filepaths_in_iterable_exist, check_valid_dataframe, check_int
+
 
 class HeatMapperClfSingleCore(ConfigReader, PlottingMixin):
     """

simba/plotting/heat_mapper_clf_mp.py

@@ -4,7 +4,7 @@
 import multiprocessing
 import os
 import platform
-from typing import Union, List
+from typing import List, Union
 
 import cv2
 import numpy as np
@@ -13,13 +13,18 @@
 
 import simba.mixins.plotting_mixin
 from simba.mixins.config_reader import ConfigReader
-from simba.mixins.plotting_mixin import PlottingMixin
 from simba.mixins.geometry_mixin import GeometryMixin
+from simba.mixins.plotting_mixin import PlottingMixin
+from simba.utils.checks import (
+    check_all_file_names_are_represented_in_video_log,
+    check_filepaths_in_iterable_exist, check_int, check_str,
+    check_valid_boolean, check_valid_dataframe, check_valid_dict)
 from simba.utils.enums import Formats
-from simba.utils.errors import NoSpecifiedOutputError, InvalidInputError
+from simba.utils.errors import InvalidInputError, NoSpecifiedOutputError
 from simba.utils.printing import SimbaTimer, stdout_success
-from simba.utils.read_write import (concatenate_videos_in_folder, get_fn_ext, read_df, remove_a_folder, find_core_cnt)
-from simba.utils.checks import check_valid_boolean, check_int, check_str, check_valid_dict, check_filepaths_in_iterable_exist, check_all_file_names_are_represented_in_video_log, check_valid_dataframe
+from simba.utils.read_write import (concatenate_videos_in_folder,
+                                    find_core_cnt, get_fn_ext, read_df,
+                                    remove_a_folder)
 
 
 def _heatmap_multiprocessor(data: np.array,

simba/plotting/heat_mapper_location_mp.py

@@ -13,12 +13,14 @@
 from simba.mixins.plotting_mixin import PlottingMixin
 from simba.utils.checks import (
     check_all_file_names_are_represented_in_video_log,
-    check_file_exist_and_readable, check_float, check_if_keys_exist_in_dict,
-    check_int, check_valid_lst, check_filepaths_in_iterable_exist)
+    check_file_exist_and_readable, check_filepaths_in_iterable_exist,
+    check_float, check_if_keys_exist_in_dict, check_int, check_valid_lst)
 from simba.utils.enums import Defaults, Formats, TagNames
 from simba.utils.errors import NoSpecifiedOutputError
 from simba.utils.printing import SimbaTimer, log_event, stdout_success
-from simba.utils.read_write import (concatenate_videos_in_folder,find_core_cnt, get_fn_ext, read_df, remove_a_folder)
+from simba.utils.read_write import (concatenate_videos_in_folder,
+                                    find_core_cnt, get_fn_ext, read_df,
+                                    remove_a_folder)
 
 STYLE_PALETTE = 'palette'
 STYLE_SHADING = 'shading'

simba/utils/lookups.py

@@ -18,7 +18,7 @@
 import simba
 from simba.utils.checks import (check_file_exist_and_readable,
                                 check_if_dir_exists)
-from simba.utils.enums import OS, FontPaths, Methods, Paths, UML
+from simba.utils.enums import OS, UML, FontPaths, Methods, Paths
 from simba.utils.read_write import get_fn_ext
 from simba.utils.warnings import NoDataFoundWarning