simba-uw-tf-dev 4.6.3__py3-none-any.whl → 4.6.4__py3-none-any.whl

simba/assets/lookups/tooptips.json

@@ -38,5 +38,10 @@
   "EGOCENTRIC_ANCHOR": "This body-part will be placed in the center of the video",
   "EGOCENTRIC_DIRECTION_ANCHOR": "This body-part will be placed at N degrees relative to the anchor",
   "EGOCENTRIC_DIRECTION": "The anchor body-part will always be placed at these degrees relative to the center anchor",
-  "CORE_COUNT": "Higher core counts speeds up processing but may require more RAM memory"
+  "CORE_COUNT": "Higher core counts speeds up processing but may require more RAM memory",
+  "KLEINBERG_SIGMA": "Higher values (e.g., 2-3) produce fewer but longer bursts; lower values (e.g., 1.1-1.5) detect more frequent, shorter bursts. Must be > 1.01",
+  "KLEINBERG_GAMMA": "Higher values (e.g., 0.5-1.0) reduce total burst count by making downward transitions costly; lower values (e.g., 0.1-0.3) allow more flexible state changes",
+  "KLEINBERG_HIERARCHY": "Hierarchy level to extract bursts from (0=lowest, higher=more selective).\n Level 0 captures all bursts; level 1-2 typically filters noise; level 3+ selects only the most prominent, sustained bursts.\nHigher levels yield fewer but more confident detections",
+  "KLEINBERG_HIERARCHY_SEARCH": "If True, searches for target hierarchy level within detected burst periods,\n falling back to lower levels if target not found. If False, extracts only bursts at the exact specified hierarchy level.\n Recommended when target hierarchy may be sparse.",
+  "KLEINBERG_SAVE_ORIGINALS": "If True, saves the original data in a new sub-directory of \nthe project_folder/csv/machine_results directory"
 }

simba/data_processors/kleinberg_calculator.py

@@ -13,10 +13,10 @@ from simba.data_processors.pybursts_calculator import kleinberg_burst_detection
 from simba.mixins.config_reader import ConfigReader
 from simba.utils.checks import (check_float, check_if_dir_exists,
                                 check_if_filepath_list_is_empty, check_int,
-                                check_that_column_exist, check_valid_lst)
+                                check_that_column_exist, check_valid_lst, check_valid_boolean)
 from simba.utils.enums import Paths, TagNames
 from simba.utils.printing import SimbaTimer, log_event, stdout_success
-from simba.utils.read_write import get_fn_ext, read_df, write_df
+from simba.utils.read_write import get_fn_ext, read_df, write_df, get_current_time, find_files_of_filetypes_in_directory, remove_a_folder, copy_files_to_directory
 from simba.utils.warnings import KleinbergWarning
 
 
@@ -38,12 +38,13 @@ class KleinbergCalculator(ConfigReader):
 
     :param str config_path: path to SimBA project config file in Configparser format
     :param List[str] classifier_names: Classifier names to apply Kleinberg smoothing to.
-    :param float sigma: Burst detection sigma value. Higher sigma values and fewer, longer, behavioural bursts will be recognised. Default: 2.
-    :param float gamma: Burst detection gamma value. Higher gamma values and fewer behavioural bursts will be recognised. Default: 0.3.
-    :param int hierarchy: Burst detection hierarchy level. Higher hierarchy values and fewer behavioural bursts will to be recognised. Default: 1.
-    :param bool hierarchical_search: See `Tutorial <https://github.com/sgoldenlab/simba/blob/master/docs/kleinberg_filter.md#hierarchical-search-example>`_ Default: False.
+    :param float sigma: State transition cost for moving to higher burst levels. Higher values (e.g., 2-3) produce fewer but longer bursts; lower values (e.g., 1.1-1.5) detect more frequent, shorter bursts. Must be > 1.01. Default: 2.
+    :param float gamma: State transition cost for moving to lower burst levels. Higher values (e.g., 0.5-1.0) reduce total burst count by making downward transitions costly; lower values (e.g., 0.1-0.3) allow more flexible state changes. Must be >= 0. Default: 0.3.
+    :param int hierarchy: Hierarchy level to extract bursts from (0=lowest, higher=more selective). Level 0 captures all bursts; level 1-2 typically filters noise; level 3+ selects only the most prominent, sustained bursts. Higher levels yield fewer but more confident detections. Must be >= 0. Default: 1.
+    :param bool hierarchical_search: If True, searches for target hierarchy level within detected burst periods, falling back to lower levels if target not found. If False, extracts only bursts at the exact specified hierarchy level. Recommended when target hierarchy may be sparse. Default: False.
     :param Optional[Union[str, os.PathLike]] input_dir: The directory with files to perform kleinberg smoothing on. If None, defaults to `project_folder/csv/machine_results`
     :param Optional[Union[str, os.PathLike]] output_dir: Location to save smoothened data in. If None, defaults to `project_folder/csv/machine_results`
+    :param Optional[bool] save_originals: If True, saves the original data in sub-directory of the ouput directory.`
 
     :example I:
     >>> kleinberg_calculator = KleinbergCalculator(config_path='MySimBAConfigPath', classifier_names=['Attack'], sigma=2, gamma=0.3, hierarchy=2, hierarchical_search=False)
@@ -68,10 +69,12 @@ class KleinbergCalculator(ConfigReader):
 
     def __init__(self,
                  config_path: Union[str, os.PathLike],
-                 classifier_names: List[str],
-                 sigma: Optional[int] = 2,
-                 gamma: Optional[float] = 0.3,
+                 classifier_names: Optional[List[str]] = None,
+                 sigma: float = 2,
+                 gamma: float = 0.3,
                  hierarchy: Optional[int] = 1,
+                 verbose: bool = True,
+                 save_originals: bool = True,
                  hierarchical_search: Optional[bool] = False,
                  input_dir: Optional[Union[str, os.PathLike]] = None,
                  output_dir: Optional[Union[str, os.PathLike]] = None):
@@ -81,25 +84,31 @@ class KleinbergCalculator(ConfigReader):
         check_float(value=sigma, name=f'{self.__class__.__name__} sigma', min_value=1.01)
         check_float(value=gamma, name=f'{self.__class__.__name__} gamma', min_value=0)
         check_int(value=hierarchy, name=f'{self.__class__.__name__} hierarchy', min_value=0)
-        check_valid_lst(data=classifier_names, source=f'{self.__class__.__name__} classifier_names', valid_dtypes=(str,), min_len=1)
+        if isinstance(classifier_names, list):
+            check_valid_lst(data=classifier_names, source=f'{self.__class__.__name__} classifier_names', valid_dtypes=(str,), min_len=1)
+        else:
+            classifier_names = deepcopy(self.clf_names)
+        check_valid_boolean(value=verbose, source=f'{self.__class__.__name__} verbose', raise_error=True)
+        check_valid_boolean(value=save_originals, source=f'{self.__class__.__name__} save_originals', raise_error=True)
         self.hierarchical_search, sigma, gamma, hierarchy, self.output_dir = (hierarchical_search, float(sigma), float(gamma), int(hierarchy), output_dir)
-        self.sigma, self.gamma, self.hierarchy, self.clfs = ( float(sigma), float(gamma), float(hierarchy), classifier_names)
+        self.sigma, self.gamma, self.hierarchy, self.clfs = ( float(sigma), float(gamma), int(hierarchy), classifier_names)
+        self.verbose, self.save_originals = verbose, save_originals
         if input_dir is None:
-            self.data_paths, self.output_dir = self.machine_results_paths, self.machine_results_dir
-            check_if_filepath_list_is_empty(filepaths=self.machine_results_paths, error_msg=f"SIMBA ERROR: No data files found in {self.machine_results_dir}. Cannot perform Kleinberg smoothing")
-            original_data_files_folder = os.path.join(self.project_path, Paths.MACHINE_RESULTS_DIR.value, f"Pre_Kleinberg_{self.datetime}")
-            if not os.path.exists(original_data_files_folder):
-                os.makedirs(original_data_files_folder)
-            for file_path in self.machine_results_paths:
-                _, file_name, ext = get_fn_ext(file_path)
-                shutil.copyfile(file_path, os.path.join(original_data_files_folder, file_name + ext))
+            self.input_dir = os.path.join(self.project_path, Paths.MACHINE_RESULTS_DIR.value)
         else:
             check_if_dir_exists(in_dir=input_dir)
-            self.data_paths = glob.glob(input_dir + f"/*.{self.file_type}")
-            check_if_filepath_list_is_empty(filepaths=self.data_paths, error_msg=f"SIMBA ERROR: No data files found in {input_dir}. Cannot perform Kleinberg smoothing")
-            if not os.path.isdir(output_dir):
-                os.makedirs(output_dir)
-        print(f"Processing Kleinberg burst detection for {len(self.data_paths)} file(s) and {len(classifier_names)} classifier(s)...")
+            self.input_dir = deepcopy(input_dir)
+        self.data_paths = find_files_of_filetypes_in_directory(directory=self.input_dir, extensions=[f'.{self.file_type}'], sort_alphabetically=True, raise_error=True)
+        if output_dir is None:
+            self.output_dir = deepcopy(self.input_dir)
+        else:
+            check_if_dir_exists(in_dir=output_dir)
+            self.output_dir = deepcopy(output_dir)
+        self.original_data_files_folder = os.path.join(self.output_dir, f"Pre_Kleinberg_{self.datetime}")
+        remove_a_folder(folder_dir=self.original_data_files_folder, ignore_errors=True)
+        os.makedirs(self.original_data_files_folder)
+        copy_files_to_directory(file_paths=self.data_paths, dir=self.original_data_files_folder, verbose=False, integer_save_names=False)
+        if self.verbose: print(f"Processing Kleinberg burst detection for {len(self.data_paths)} file(s) and {len(classifier_names)} classifier(s)...")
 
     def hierarchical_searcher(self):
         if (len(self.kleinberg_bouts["Hierarchy"]) == 1) and (int(self.kleinberg_bouts.at[0, "Hierarchy"]) == 0):
@@ -135,7 +144,7 @@ class KleinbergCalculator(ConfigReader):
         for file_cnt, file_path in enumerate(self.data_paths):
             _, video_name, _ = get_fn_ext(file_path)
             video_timer = SimbaTimer(start=True)
-            print(f"Performing Kleinberg burst detection for video {video_name}  (Video {file_cnt+1}/{len(self.data_paths)})...")
+            if self.verbose: print(f"[{get_current_time()}] Performing Kleinberg burst detection for video {video_name} (Video {file_cnt+1}/{len(self.data_paths)})...")
             data_df = read_df(file_path, self.file_type).reset_index(drop=True)
             video_out_df = deepcopy(data_df)
             check_that_column_exist(df=data_df, column_name=self.clfs, file_name=video_name)
@@ -150,7 +159,7 @@ class KleinbergCalculator(ConfigReader):
                     self.kleinberg_bouts.insert(loc=0, column="Video", value=video_name)
                     detailed_df_lst.append(self.kleinberg_bouts)
                     if self.hierarchical_search:
-                        print(f"Applying hierarchical search for video {video_name}...")
+                        if self.verbose: print(f"[{get_current_time()}] Applying hierarchical search for video {video_name}...")
                         self.hierarchical_searcher()
                     else:
                         self.clf_bouts_in_hierarchy = self.kleinberg_bouts[self.kleinberg_bouts["Hierarchy"] == self.hierarchy]
@@ -160,19 +169,38 @@ class KleinbergCalculator(ConfigReader):
                     video_out_df.loc[hierarchy_idx, clf] = 1
             write_df(video_out_df, self.file_type, save_path)
             video_timer.stop_timer()
-            print(f'Kleinberg analysis complete for video {video_name} (saved at {save_path}), elapsed time: {video_timer.elapsed_time_str}s.')
+            if self.verbose: print(f'[{get_current_time()}] Kleinberg analysis complete for video {video_name} (saved at {save_path}), elapsed time: {video_timer.elapsed_time_str}s.')
 
         self.timer.stop_timer()
+        if not self.save_originals:
+            remove_a_folder(folder_dir=self.original_data_files_folder, ignore_errors=False)
+        else:
+            if self.verbose: stdout_success(msg=f"Original, un-smoothened data, saved in {self.original_data_files_folder} directory", elapsed_time=self.timer.elapsed_time_str, source=self.__class__.__name__)
         if len(detailed_df_lst) > 0:
             self.detailed_df = pd.concat(detailed_df_lst, axis=0)
             detailed_save_path = os.path.join(self.logs_path, f"Kleinberg_detailed_log_{self.datetime}.csv")
             self.detailed_df.to_csv(detailed_save_path)
-            stdout_success(msg=f"Kleinberg analysis complete. See {detailed_save_path} for details of detected bouts of all classifiers in all hierarchies", elapsed_time=self.timer.elapsed_time_str, source=self.__class__.__name__)
+            if self.verbose:  stdout_success(msg=f"Kleinberg analysis complete for {len(self.data_paths)} files. Results stored in {self.output_dir} directory. See {detailed_save_path} for details of detected bouts of all classifiers in all hierarchies", elapsed_time=self.timer.elapsed_time_str, source=self.__class__.__name__)
         else:
-            print("Kleinberg analysis complete.")
+            if self.verbose: print(f"[{get_current_time()}] Kleinberg analysis complete for {len(self.data_paths)} files. Results stored in {self.output_dir} directory.")
             KleinbergWarning(msg="All behavior bouts removed following kleinberg smoothing", source=self.__class__.__name__)
 
 
+
+
+# test = KleinbergCalculator(config_path=r"C:\troubleshooting\mitra\project_folder\project_config.ini",
+#                            classifier_names=['straub_tail'],
+#                            sigma=1.1,
+#                            gamma=0.1,
+#                            hierarchy=1,
+#                            save_originals=False,
+#                            hierarchical_search=False)
+#
+# test.run()
+#
+
+
+
 # test = KleinbergCalculator(config_path='/Users/simon/Desktop/envs/simba/troubleshooting/levi/project_folder/project_config.ini',
 #                            classifier_names=['No_Seizure_(0)'],
 #                            sigma=1.1,

simba/plotting/gantt_creator_mp.py

@@ -4,17 +4,18 @@ import warnings
 
 warnings.simplefilter(action="ignore", category=FutureWarning)
 import functools
+import gc
 import multiprocessing
 import os
 import platform
-from typing import List, Optional, Union
 from copy import deepcopy
+from typing import List, Optional, Union
 
-import gc
 import cv2
+import matplotlib
 import numpy as np
 import pandas as pd
-import matplotlib
+
 matplotlib.use('Agg')
 
 from simba.mixins.config_reader import ConfigReader
@@ -23,13 +24,14 @@ from simba.utils.checks import (
     check_all_file_names_are_represented_in_video_log,
     check_file_exist_and_readable, check_int, check_str,
     check_that_column_exist, check_valid_boolean, check_valid_lst)
-from simba.utils.data import (create_color_palette, detect_bouts, terminate_cpu_pool, get_cpu_pool)
+from simba.utils.data import (create_color_palette, detect_bouts, get_cpu_pool,
+                              terminate_cpu_pool)
 from simba.utils.enums import Formats, Options
 from simba.utils.errors import NoSpecifiedOutputError
 from simba.utils.printing import SimbaTimer, stdout_success
 from simba.utils.read_write import (concatenate_videos_in_folder,
                                     create_directory, find_core_cnt,
-                                    get_fn_ext, read_df, get_current_time)
+                                    get_current_time, get_fn_ext, read_df)
 
 HEIGHT = "height"
 WIDTH = "width"

simba/plotting/pose_plotter_mp.py

@@ -12,15 +12,19 @@ import pandas as pd
 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_instance, check_int, check_nvidea_gpu_available, check_str, check_that_column_exist, check_valid_boolean)
-from simba.utils.data import create_color_palette, terminate_cpu_pool, get_cpu_pool
+from simba.utils.checks import (check_instance, check_int,
+                                check_nvidea_gpu_available, check_str,
+                                check_that_column_exist, check_valid_boolean)
+from simba.utils.data import (create_color_palette, get_cpu_pool,
+                              terminate_cpu_pool)
 from simba.utils.enums import OS, Formats, Options
 from simba.utils.errors import CountError, InvalidFilepathError
 from simba.utils.printing import SimbaTimer, stdout_success
 from simba.utils.read_write import (concatenate_videos_in_folder,
                                     find_core_cnt,
                                     find_files_of_filetypes_in_directory,
-                                    get_fn_ext, get_video_meta_data, read_df, get_current_time)
+                                    get_current_time, get_fn_ext,
+                                    get_video_meta_data, read_df)
 from simba.utils.warnings import FrameRangeWarning
 
 

simba/plotting/roi_plotter_mp.py

@@ -24,15 +24,16 @@ from simba.utils.checks import (check_file_exist_and_readable, check_float,
                                 check_valid_boolean, check_valid_lst,
                                 check_video_and_data_frm_count_align)
 from simba.utils.data import (create_color_palettes, detect_bouts,
-                              slice_roi_dict_for_video, terminate_cpu_pool, get_cpu_pool)
+                              get_cpu_pool, slice_roi_dict_for_video,
+                              terminate_cpu_pool)
 from simba.utils.enums import ROI_SETTINGS, Formats, Keys, Paths, TextOptions
 from simba.utils.errors import (BodypartColumnNotFoundError, DuplicationError,
                                 NoFilesFoundError, NoROIDataError,
                                 ROICoordinatesNotFoundError)
 from simba.utils.printing import SimbaTimer, stdout_success
 from simba.utils.read_write import (concatenate_videos_in_folder,
-                                    find_core_cnt, get_video_meta_data,
-                                    read_df, get_current_time)
+                                    find_core_cnt, get_current_time,
+                                    get_video_meta_data, read_df)
 from simba.utils.warnings import (DuplicateNamesWarning, FrameRangeWarning,
                                   GPUToolsWarning)
 

simba/plotting/yolo_pose_track_visualizer.py

@@ -13,10 +13,11 @@ from simba.mixins.plotting_mixin import PlottingMixin
 from simba.utils.checks import (check_file_exist_and_readable, check_float,
                                 check_if_dir_exists, check_int,
                                 check_valid_boolean, check_valid_dataframe)
-from simba.utils.data import terminate_cpu_pool, get_cpu_pool
+from simba.utils.data import get_cpu_pool, terminate_cpu_pool
 from simba.utils.enums import Defaults, Formats
 from simba.utils.errors import InvalidFilepathError, NoFilesFoundError
-from simba.utils.lookups import get_random_color_palette, intermittent_palette, get_current_time
+from simba.utils.lookups import (get_current_time, get_random_color_palette,
+                                 intermittent_palette)
 from simba.utils.printing import SimbaTimer, stdout_success
 from simba.utils.read_write import (concatenate_videos_in_folder,
                                     create_directory,

simba/plotting/yolo_pose_visualizer.py

@@ -14,7 +14,8 @@ from simba.utils.checks import (check_file_exist_and_readable, check_float,
                                 check_if_dir_exists, check_int,
                                 check_valid_boolean, check_valid_dataframe,
                                 check_valid_lst, check_valid_tuple)
-from simba.utils.data import create_color_palette, terminate_cpu_pool, get_cpu_pool
+from simba.utils.data import (create_color_palette, get_cpu_pool,
+                              terminate_cpu_pool)
 from simba.utils.enums import Defaults, Options
 from simba.utils.errors import (CountError, DataHeaderError, FrameRangeError,
                                 InvalidInputError, NoDataError)
@@ -22,9 +23,9 @@ from simba.utils.printing import SimbaTimer, stdout_success
 from simba.utils.read_write import (concatenate_videos_in_folder,
                                     create_directory, find_core_cnt,
                                     find_files_of_filetypes_in_directory,
-                                    get_fn_ext, get_video_meta_data,
-                                    read_frm_of_video, recursive_file_search,
-                                    remove_a_folder, get_current_time)
+                                    get_current_time, get_fn_ext,
+                                    get_video_meta_data, read_frm_of_video,
+                                    recursive_file_search, remove_a_folder)
 
 FRAME = 'FRAME'
 CLASS_ID = 'CLASS_ID'

simba/ui/pop_ups/kleinberg_pop_up.py

@@ -7,13 +7,13 @@ from typing import Union
 from simba.data_processors.kleinberg_calculator import KleinbergCalculator
 from simba.mixins.config_reader import ConfigReader
 from simba.mixins.pop_up_mixin import PopUpMixin
-from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, Entry_Box,
-                                        SimbaButton, SimBADropDown)
+from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, Entry_Box, SimbaButton, SimBADropDown, SimBALabel)
 from simba.utils.checks import check_float, check_int
 from simba.utils.enums import Formats, Keys, Links
 from simba.utils.errors import NoChoosenClassifierError, NoDataError
-from simba.utils.read_write import str_2_bool
+from simba.utils.read_write import str_2_bool, get_current_time
 
+INSTRUCTIONS_TXT = 'Results in the project_folder/csv/machine_results folder are overwritten.\n If saving the originals, the original un-smoothened data is saved in a subdirectory of \nthe project_folder/csv/machine_results folder'
 
 class KleinbergPopUp(PopUpMixin, ConfigReader):
     def __init__(self,
@@ -24,61 +24,59 @@ class KleinbergPopUp(PopUpMixin, ConfigReader):
             raise NoDataError(msg=f'Cannot perform Kleinberg smoothing: No data files found in {self.machine_results_dir} directory', source=self.__class__.__name__)
         PopUpMixin.__init__(self, title="APPLY KLEINBERG BEHAVIOR CLASSIFICATION SMOOTHING", icon='smooth')
         kleinberg_settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="KLEINBERG SETTINGS", icon_name=Keys.DOCUMENTATION.value, icon_link=Links.KLEINBERG.value)
-        self.k_sigma = Entry_Box(kleinberg_settings_frm, fileDescription="SIGMA", img='sigma', value='2', justify='center', labelwidth=35, entry_box_width=35)
-        self.k_gamma = Entry_Box(kleinberg_settings_frm, fileDescription="GAMMA", img='gamma', value='0.3', justify='center', labelwidth=35, entry_box_width=35)
-        self.k_hierarchy = Entry_Box(kleinberg_settings_frm, fileDescription="HIERARCHY", value=1, img='hierarchy_2', justify='center', labelwidth=35, entry_box_width=35)
-        self.h_search_dropdown = SimBADropDown(parent=kleinberg_settings_frm, dropdown_options=['TRUE', 'FALSE'], label="HIERACHICAL SEARCH", value='FALSE', img='hierarchy', label_width=35, dropdown_width=35)
+        self.k_sigma = Entry_Box(kleinberg_settings_frm, fileDescription="SIGMA", img='sigma', value='2', justify='center', labelwidth=35, entry_box_width=35, tooltip_key='KLEINBERG_SIGMA')
+        self.k_gamma = Entry_Box(kleinberg_settings_frm, fileDescription="GAMMA", img='gamma', value='0.3', justify='center', labelwidth=35, entry_box_width=35, tooltip_key='KLEINBERG_GAMMA')
+        self.k_hierarchy = Entry_Box(kleinberg_settings_frm, fileDescription="HIERARCHY", value=1, img='hierarchy_2', justify='center', labelwidth=35, entry_box_width=35, validation='numeric', tooltip_key='KLEINBERG_HIERARCHY')
+        self.h_search_dropdown = SimBADropDown(parent=kleinberg_settings_frm, dropdown_options=['TRUE', 'FALSE'], label="HIERARCHICAL SEARCH", value='FALSE', img='hierarchy', label_width=35, dropdown_width=35, tooltip_key='KLEINBERG_HIERARCHY_SEARCH')
+        self.save_originals_dropdown = SimBADropDown(parent=kleinberg_settings_frm, dropdown_options=['TRUE', 'FALSE'], label="SAVE ORIGINAL DATA:", value='TRUE', img='save', label_width=35, dropdown_width=35, tooltip_key='KLEINBERG_SAVE_ORIGINALS')
+        self.instructions_lbl = SimBALabel(parent=kleinberg_settings_frm, txt=INSTRUCTIONS_TXT, justify='center', txt_clr='blue', font=Formats.FONT_REGULAR_ITALICS.value)
 
 
-        kleinberg_table_frame = LabelFrame(self.main_frm, text="CHOOSE CLASSIFIER(S) FOR KLEINBERG SMOOTHING", font=Formats.FONT_HEADER.value)
+        kleinberg_table_frame = CreateLabelFrameWithIcon(parent=self.main_frm, header="CHOOSE CLASSIFIER(S) FOR KLEINBERG SMOOTHING", icon_name=Keys.DOCUMENTATION.value, icon_link=Links.KLEINBERG.value)
         clf_var_dict, clf_cb_dict = {}, {}
         for clf_cnt, clf in enumerate(self.clf_names):
             clf_var_dict[clf] = BooleanVar()
             clf_cb_dict[clf] = Checkbutton(kleinberg_table_frame, text=clf, font=Formats.FONT_REGULAR.value, variable=clf_var_dict[clf])
-            clf_cb_dict[clf].grid(row=clf_cnt, sticky=NW)
+            clf_cb_dict[clf].grid(row=clf_cnt, column=0, sticky=NW)
 
         run_kleinberg_btn = SimbaButton(parent=self.main_frm, txt="APPLY KLEINBERG SMOOTHER", img='rocket', txt_clr="blue", font=Formats.FONT_REGULAR.value, cmd=self.run_kleinberg, cmd_kwargs={'behaviors_dict': lambda: clf_var_dict, 'hierarchical_search': lambda: str_2_bool(self.h_search_dropdown.get_value())})
-        kleinberg_settings_frm.grid(row=0, sticky=W, padx=10)
-        self.k_sigma.grid(row=0, sticky=W)
-        self.k_gamma.grid(row=1, sticky=W)
-        self.k_hierarchy.grid(row=2, sticky=W)
-        self.h_search_dropdown.grid(row=3, column=0, sticky=W)
-        kleinberg_table_frame.grid(row=1, pady=10, padx=10)
-        run_kleinberg_btn.grid(row=2)
+        kleinberg_settings_frm.grid(row=0, sticky=W, pady=(15, 0))
+        self.instructions_lbl.grid(row=0, sticky=W)
+        self.k_sigma.grid(row=1, sticky=W)
+        self.k_gamma.grid(row=2, sticky=W)
+        self.k_hierarchy.grid(row=3, sticky=W)
+        self.h_search_dropdown.grid(row=4, column=0, sticky=W)
+        self.save_originals_dropdown.grid(row=5, column=0, sticky=W)
+        kleinberg_table_frame.grid(row=1, column=0, sticky=NW, pady=(15, 0))
+        run_kleinberg_btn.grid(row=2, column=0, sticky=NW, pady=(15, 0))
         self.main_frm.mainloop()
 
     def run_kleinberg(self, behaviors_dict: dict, hierarchical_search: bool):
         targets = []
         for behaviour, behavior_val in behaviors_dict.items():
-            if behavior_val.get():
-                targets.append(behaviour)
+            if behavior_val.get(): targets.append(behaviour)
 
         if len(targets) == 0:
             raise NoChoosenClassifierError(source=self.__class__.__name__)
 
-        check_int(name="Hierarchy", value=self.k_hierarchy.entry_get)
-        check_float(name="Sigma", value=self.k_sigma.entry_get)
-        check_float(name="Gamma", value=self.k_gamma.entry_get)
-
-        try:
-            print(
-                "Applying kleinberg hyperparameter Setting: Sigma: {}, Gamma: {}, Hierarchy: {}".format(
-                    str(self.k_sigma.entry_get),
-                    str(self.k_gamma.entry_get),
-                    str(self.k_hierarchy.entry_get),
-                )
-            )
-        except:
-            print("Please insert accurate values for all hyperparameters.")
-
-        kleinberg_analyzer = KleinbergCalculator(
-            config_path=self.config_path,
-            classifier_names=targets,
-            sigma=self.k_sigma.entry_get,
-            gamma=self.k_gamma.entry_get,
-            hierarchy=self.k_hierarchy.entry_get,
-            hierarchical_search=hierarchical_search,
-        )
+        k_hierarchy = self.k_hierarchy.entry_get
+        k_sigma = self.k_sigma.entry_get
+        k_gamma = self.k_gamma.entry_get
+        save_originals = str_2_bool(self.save_originals_dropdown.get_value())
+
+        check_int(name="Hierarchy", value=k_hierarchy, min_value=1, allow_negative=False, allow_zero=False, raise_error=True)
+        check_float(name="Sigma", value=k_sigma, allow_negative=False, allow_zero=False, raise_error=True)
+        check_float(name="Gamma", value=k_gamma, allow_negative=False, allow_zero=False, raise_error=True)
+
+        print(f"[{get_current_time()}] Applying kleinberg hyperparameter Setting: Sigma: {k_sigma}, Gamma: {k_gamma}, Hierarchy: {k_hierarchy}")
+
+        kleinberg_analyzer = KleinbergCalculator(config_path=self.config_path,
+                                                 classifier_names=targets,
+                                                 sigma=float(k_sigma),
+                                                 gamma=float(k_gamma),
+                                                 hierarchy=int(k_hierarchy),
+                                                 hierarchical_search=hierarchical_search,
+                                                 save_originals=save_originals)
         kleinberg_analyzer.run()
 
 

simba/ui/tkinter_functions.py

@@ -229,6 +229,7 @@ class Entry_Box(Frame):
                  value: Optional[Any] = None,
                  label_font: tuple = Formats.FONT_REGULAR.value,
                  entry_font: tuple = Formats.FONT_REGULAR.value,
+                 tooltip_key: Optional[str] = None,
                  justify: Literal["left", "center", "right"] = 'left',
                  cmd: Optional[Callable] = None,
                  **kw):
@@ -249,6 +250,8 @@ class Entry_Box(Frame):
         self.filePath = StringVar()
         self.lblName = Label(self, text=fileDescription, width=labelwidth, anchor=W, font=label_font, bg=label_bg_clr)
         self.lblName.grid(row=0, column=1)
+        if tooltip_key in TOOLTIPS.keys():
+            CreateToolTip(widget=self.lblName, text=TOOLTIPS[tooltip_key])
         if not entry_box_width:
             self.entPath = Entry(self, textvariable=self.filePath, state=self.status,  validate="key", validatecommand=self.validation_methods.get(validation, None), font=entry_font, justify=justify, bg=entry_box_clr)
         else:

simba/utils/printing.py

@@ -1,125 +1,124 @@
-__author__ = "Simon Nilsson; sronilsson@gmail.com"
-
-try:
-    from typing import Literal
-except:
-    from typing_extensions import Literal
-
-
-import logging
-import time
-from typing import Optional
-from datetime import datetime
-
-from simba.utils.enums import Defaults, TagNames
-
-
-def stdout_success(msg: str, source: Optional[str] = "", elapsed_time: Optional[str] = None) -> None:
-    """
-    Helper to parse msg of completed operation to SimBA main interface.
-
-    :param str msg: Message to be parsed.
-    :param Optional[str] source: Optional string indicating the source method or function of the msg for logging.
-    :param Optional[str] elapsed_time: Optional string indicating the runtime of the completed operation.
-    :return None:
-    """
-
-    log_event(logger_name=f"{source}.{stdout_success.__name__}", log_type=TagNames.COMPLETE.value, msg=msg)
-    if elapsed_time:
-        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA COMPLETE: {msg} (elapsed time: {elapsed_time}s) {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.COMPLETE.value}")
-    else:
-        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA COMPLETE: {msg} {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.COMPLETE.value}")
-
-
-def stdout_warning(msg: str, elapsed_time: Optional[str] = None) -> None:
-    """
-    Helper to parse warning msg to SimBA main interface.
-
-    :param str msg: Message to be parsed.
-    :param Optional[str] source: Optional string indicating the source method or function of the msg for logging.
-    :param elapsed_time: Optional string indicating the runtime.
-    :return None:
-    """
-
-    if elapsed_time:
-        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA WARNING: {msg} (elapsed time: {elapsed_time}s) {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.WARNING.value}")
-    else:
-        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA WARNING: {msg} {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.WARNING.value}")
-
-
-def stdout_trash(msg: str, source: Optional[str] = "", elapsed_time: Optional[str] = None) -> None:
-    """
-    Helper to parse msg of delete operation to SimBA main interface.
-
-    :param str msg: Message to be parsed.
-    :param Optional[str] source: Optional string indicating the source method or function of the operation for logging.
-    :param elapsed_time: Optional string indicating the runtime.
-    :return None:
-    """
-
-    log_event(logger_name=f"{source}.{stdout_trash.__name__}", log_type=TagNames.TRASH.value, msg=msg)
-    if elapsed_time:
-        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA COMPLETE: {msg} (elapsed time: {elapsed_time}s) {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.TRASH.value}")
-    else:
-        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA COMPLETE: {msg} {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.TRASH.value}")
-
-
-def stdout_information(msg: str, source: Optional[str] = "", elapsed_time: Optional[str] = None) -> None:
-    """
-    Helper to parse information msg to SimBA main interface. E.g., how many monitors and their resolutions which is available.
-
-    :param str msg: Message to be parsed.
-    :param Optional[str] source: Optional string indicating the source method or function of the operation for logging.
-    :param elapsed_time: Optional string indicating the runtime.
-    :return None:
-    """
-
-    log_event(logger_name=f"{source}.{stdout_trash.__name__}", log_type=TagNames.INFORMATION.value, msg=msg)
-    if elapsed_time:
-        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA COMPLETE: {msg} (elapsed time: {elapsed_time}s) {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.INFORMATION.value}")
-    else:
-        print(f"[{datetime.now().strftime('%H:%M:%S')}] {msg} {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.INFORMATION.value}")
-
-
-class SimbaTimer(object):
-    """Timer class for keeping track of start and end-times of calls"""
-
-    def __init__(self, start: bool = False):
-        if start:
-            self.start_timer()
-
-    def start_timer(self):
-        self.timer = time.time()
-
-    def stop_timer(self):
-        if not hasattr(self, "timer"):
-            self.elapsed_time = -1
-            self.elapsed_time_str = "-1"
-        else:
-            self.elapsed_time = round(time.time() - self.timer, 4)
-            self.elapsed_time_str = str(self.elapsed_time)
-
-
-def log_event(logger_name: str, log_type: Literal["CLASS_INIT", "error", "warning"], msg: str):
-    logger = logging.getLogger(str(logger_name))
-    if log_type == TagNames.CLASS_INIT.value:
-        logger.info(f"{TagNames.CLASS_INIT.value}||{msg}")
-    elif log_type == TagNames.ERROR.value:
-        logger.error(f"{TagNames.ERROR.value}||{msg}")
-    elif log_type == TagNames.WARNING.value:
-        logger.warning(f"{TagNames.WARNING.value}||{msg}")
-    elif log_type == TagNames.TRASH.value:
-        logger.info(f"{TagNames.TRASH.value}||{msg}")
-    elif log_type == TagNames.COMPLETE.value:
-        logger.info(f"{TagNames.COMPLETE.value}||{msg}")
-
-
-def perform_timing(func):
-    def decorator(*args, **kwargs):
-        timer = SimbaTimer(start=True)
-        results = func(*args, **kwargs, _timer=timer)
-        timer.stop_timer()
-        results["timer"] = timer.elapsed_time_str
-        return results
-
-    return decorator
+__author__ = "Simon Nilsson; sronilsson@gmail.com"
+
+try:
+    from typing import Literal
+except:
+    from typing_extensions import Literal
+
+import logging
+import time
+from datetime import datetime
+from typing import Optional
+
+from simba.utils.enums import Defaults, TagNames
+
+
+def stdout_success(msg: str, source: Optional[str] = "", elapsed_time: Optional[str] = None) -> None:
+    """
+    Helper to parse msg of completed operation to SimBA main interface.
+
+    :param str msg: Message to be parsed.
+    :param Optional[str] source: Optional string indicating the source method or function of the msg for logging.
+    :param Optional[str] elapsed_time: Optional string indicating the runtime of the completed operation.
+    :return None:
+    """
+
+    log_event(logger_name=f"{source}.{stdout_success.__name__}", log_type=TagNames.COMPLETE.value, msg=msg)
+    if elapsed_time:
+        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA COMPLETE: {msg} (elapsed time: {elapsed_time}s) {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.COMPLETE.value}")
+    else:
+        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA COMPLETE: {msg} {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.COMPLETE.value}")
+
+
+def stdout_warning(msg: str, elapsed_time: Optional[str] = None) -> None:
+    """
+    Helper to parse warning msg to SimBA main interface.
+
+    :param str msg: Message to be parsed.
+    :param Optional[str] source: Optional string indicating the source method or function of the msg for logging.
+    :param elapsed_time: Optional string indicating the runtime.
+    :return None:
+    """
+
+    if elapsed_time:
+        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA WARNING: {msg} (elapsed time: {elapsed_time}s) {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.WARNING.value}")
+    else:
+        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA WARNING: {msg} {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.WARNING.value}")
+
+
+def stdout_trash(msg: str, source: Optional[str] = "", elapsed_time: Optional[str] = None) -> None:
+    """
+    Helper to parse msg of delete operation to SimBA main interface.
+
+    :param str msg: Message to be parsed.
+    :param Optional[str] source: Optional string indicating the source method or function of the operation for logging.
+    :param elapsed_time: Optional string indicating the runtime.
+    :return None:
+    """
+
+    log_event(logger_name=f"{source}.{stdout_trash.__name__}", log_type=TagNames.TRASH.value, msg=msg)
+    if elapsed_time:
+        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA COMPLETE: {msg} (elapsed time: {elapsed_time}s) {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.TRASH.value}")
+    else:
+        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA COMPLETE: {msg} {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.TRASH.value}")
+
+
+def stdout_information(msg: str, source: Optional[str] = "", elapsed_time: Optional[str] = None) -> None:
+    """
+    Helper to parse information msg to SimBA main interface. E.g., how many monitors and their resolutions which is available.
+
+    :param str msg: Message to be parsed.
+    :param Optional[str] source: Optional string indicating the source method or function of the operation for logging.
+    :param elapsed_time: Optional string indicating the runtime.
+    :return None:
+    """
+
+    log_event(logger_name=f"{source}.{stdout_trash.__name__}", log_type=TagNames.INFORMATION.value, msg=msg)
+    if elapsed_time:
+        print(f"[{datetime.now().strftime('%H:%M:%S')}] SIMBA COMPLETE: {msg} (elapsed time: {elapsed_time}s) {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.INFORMATION.value}")
+    else:
+        print(f"[{datetime.now().strftime('%H:%M:%S')}] {msg} {Defaults.STR_SPLIT_DELIMITER.value}{TagNames.INFORMATION.value}")
+
+
+class SimbaTimer(object):
+    """Timer class for keeping track of start and end-times of calls"""
+
+    def __init__(self, start: bool = False):
+        if start:
+            self.start_timer()
+
+    def start_timer(self):
+        self.timer = time.time()
+
+    def stop_timer(self):
+        if not hasattr(self, "timer"):
+            self.elapsed_time = -1
+            self.elapsed_time_str = "-1"
+        else:
+            self.elapsed_time = round(time.time() - self.timer, 4)
+            self.elapsed_time_str = str(self.elapsed_time)
+
+
+def log_event(logger_name: str, log_type: Literal["CLASS_INIT", "error", "warning"], msg: str):
+    logger = logging.getLogger(str(logger_name))
+    if log_type == TagNames.CLASS_INIT.value:
+        logger.info(f"{TagNames.CLASS_INIT.value}||{msg}")
+    elif log_type == TagNames.ERROR.value:
+        logger.error(f"{TagNames.ERROR.value}||{msg}")
+    elif log_type == TagNames.WARNING.value:
+        logger.warning(f"{TagNames.WARNING.value}||{msg}")
+    elif log_type == TagNames.TRASH.value:
+        logger.info(f"{TagNames.TRASH.value}||{msg}")
+    elif log_type == TagNames.COMPLETE.value:
+        logger.info(f"{TagNames.COMPLETE.value}||{msg}")
+
+
+def perform_timing(func):
+    def decorator(*args, **kwargs):
+        timer = SimbaTimer(start=True)
+        results = func(*args, **kwargs, _timer=timer)
+        timer.stop_timer()
+        results["timer"] = timer.elapsed_time_str
+        return results
+
+    return decorator

simba/utils/read_write.py

@@ -560,7 +560,11 @@ def get_video_info_ffmpeg(video_path: Union[str, os.PathLike]) -> Dict[str, Any]
 
 def remove_a_folder(folder_dir: Union[str, os.PathLike], ignore_errors: Optional[bool] = True) -> None:
     """Helper to remove a directory"""
-    check_if_dir_exists(in_dir=folder_dir, source=remove_a_folder.__name__)
+    valid_dir = check_if_dir_exists(in_dir=folder_dir, source=remove_a_folder.__name__, raise_error=False)
+    if not valid_dir and not ignore_errors:
+        raise NotDirectoryError(msg=f'Cannot delete directory {folder_dir}: The directory does not exist', source=remove_a_folder.__name__)
+    if not valid_dir and ignore_errors:
+        return
     try:
         shutil.rmtree(folder_dir, ignore_errors=ignore_errors)
     except Exception as e:

{simba_uw_tf_dev-4.6.3.dist-info → simba_uw_tf_dev-4.6.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: simba-uw-tf-dev
-Version: 4.6.3
+Version: 4.6.4
 Summary: Toolkit for computer classification and analysis of behaviors in experimental animals
 Home-page: https://github.com/sgoldenlab/simba
 Author: Simon Nilsson, Jia Jie Choong, Sophia Hwang

{simba_uw_tf_dev-4.6.3.dist-info → simba_uw_tf_dev-4.6.4.dist-info}/RECORD

@@ -458,7 +458,7 @@ simba/assets/lookups/critical_values_05.pickle,sha256=bYlqp9T6ehVIjIJr3Uwfulj-kS
 simba/assets/lookups/feature_extraction_headers.csv,sha256=I5TLfAihpHgzUZ7OUyGge-Yu-XGbQmHbDFAD4vVAc4w,2987
 simba/assets/lookups/features.csv,sha256=bc6aN2ZRa2J2dxur-Zjcpc4I3q6vd-gN2erwnhdzLIk,14175
 simba/assets/lookups/model_names.parquet,sha256=hOuvYONO8wZGcAwRNSf_hS_lUaynC8Gt24MLOg3w5ZA,270783
-simba/assets/lookups/tooptips.json,sha256=cQUtBr5-Mpy4CqFV7kV-JjLhzMZHhJWWaXZWCyac4GU,5097
+simba/assets/lookups/tooptips.json,sha256=uPFy_Ew53vTtejvup5DsmvEyfZkUeaUyGeDxXrM0crU,6156
 simba/assets/lookups/unsupervised_example_x.csv,sha256=bc6aN2ZRa2J2dxur-Zjcpc4I3q6vd-gN2erwnhdzLIk,14175
 simba/assets/lookups/yolo.yaml,sha256=3Omt-t-w555E4oAJsRRblBjzyiTW9e5x-0VuyQ-t9C0,278
 simba/assets/lookups/yolo_schematics/yolo_1.csv,sha256=TzUzWAtia4P-5FRwLPnjU0zDSkAA-nItNFCcq6nG_iQ,72
@@ -532,7 +532,7 @@ simba/data_processors/gibbs_sampler.py,sha256=Lj3lJJ0LWYLoeMRU2oniTHJSRpdG8oMIEN
 simba/data_processors/interpolate.py,sha256=GaLvrqS-iVB9TA0eEsCO3dnUCxVD_0KwWqWYZsRlyOM,8006
 simba/data_processors/interpolate_pose.py,sha256=RME5CxN-oJTiRab4iMQLMRmc7rmdsvXz490wyguDT3Q,7139
 simba/data_processors/interpolation_smoothing.py,sha256=aeN4klghvxXUDV8yu_tiXVUAV4Bxu_y0jmNGnpJe6vw,39132
-simba/data_processors/kleinberg_calculator.py,sha256=vWcyBB3MA99tdI1n2CXN_6a7CDInLyhIUnTraVhMIFM,12401
+simba/data_processors/kleinberg_calculator.py,sha256=bgTK4V83QrB9DC2rxF0YH1nwU7rF4wVvJ1WQtXA_cxU,14592
 simba/data_processors/light_dark_box_analyzer.py,sha256=FaXBkuAzI2YZToBvswCTjv1epYK6SiysQTepIQ_-qPc,9514
 simba/data_processors/movement_calculator.py,sha256=tqDzgMIH23iz_U6ABqva2VpG2Yu8GV1__DK6I6-rd7Y,12393
 simba/data_processors/mutual_exclusivity_corrector.py,sha256=XuLk8gd2oXAunpP3yJTIcfeoWofpO1iv2EFY6QcEEU8,8729
@@ -675,7 +675,7 @@ simba/plotting/distance_plotter_mp.py,sha256=7tbp63Jn9JGkox0fCaqeunE3027pEl0xI7u
 simba/plotting/ez_path_plot.py,sha256=OSQSniIKdr55iO5-BMqmrgnEOfLIjxq0-ojrJn1y7lE,10350
 simba/plotting/frame_mergerer_ffmpeg.py,sha256=AqfUQ_Lx83VfOZbz89qa1tM1WK8ETNhNITPTjnTqqeY,8485
 simba/plotting/gantt_creator.py,sha256=mbTonMIbhdHHGh9e4zzaLljp2haoKWxgwfWwZ2IVh0s,11297
-simba/plotting/gantt_creator_mp.py,sha256=wjM-NL-UCGXnl5E3Pc-RcQW4ek2pHge8ShzJCaLdgrA,16635
+simba/plotting/gantt_creator_mp.py,sha256=a5ZBWaV0pl-PcgtePBC4mZR9v6NU8WQ2waD3xBomy5k,16668
 simba/plotting/gantt_plotly.py,sha256=s727fBblZnMksHvJpmbb4_G_Eq8J3bMhlNzyL1xap_M,8406
 simba/plotting/geometry_plotter.py,sha256=Jsl_695-ZWcba9_c3MiXq6S8MufWsKUcaGbCv9HYSoo,15499
 simba/plotting/heat_mapper_clf.py,sha256=rYFWuBS3v8v14EnXhaKKzPml4lJzkPBP5FJ8RxdnKik,10675
@@ -689,18 +689,18 @@ simba/plotting/path_plotter_mp.py,sha256=1N0uR-EoSyKHvppSJNfRhLcTKKWKko_UNKRtO9r
 simba/plotting/plot_clf_results.py,sha256=MLm_S42RPvVIeng7KyrAa9z0aS5PGFSV4Pm9VCy3M18,22895
 simba/plotting/plot_clf_results_mp.py,sha256=46qiq9k5dD8wZBRfvCYThkYiPnvmeCEKZy2XEp2N3h4,32340
 simba/plotting/pose_plotter.py,sha256=aH48s-loMOy7Lyor3Z0hF6q8tgXeGBg-Mm_0ICONz2I,9145
-simba/plotting/pose_plotter_mp.py,sha256=VcWp6hP-z1tUYH0sGyIg04YzSTDH3uYLPbqeT7Dr4Xk,13667
+simba/plotting/pose_plotter_mp.py,sha256=RB-KEWuOLL4D0CGj5teqAFEWC01joqY6B_8WLSNzvuo,13803
 simba/plotting/probability_plot_creator.py,sha256=8BHYtCtPXungLxzI0ZkCKUoDhuYBJsGefsChzHwrDLI,11878
 simba/plotting/probability_plot_creator_mp.py,sha256=GVa_vaVTeBIBYHycg2UF1G4rOPvZe3BfJPl4cSdWkTs,17380
 simba/plotting/roi_plotter.py,sha256=ApMdsfeCm5iFbiCZWn7eUfpfXTTzXmcEB_j3nCbod6o,25693
-simba/plotting/roi_plotter_mp.py,sha256=xQTGa1niRqkU4EwPc6aFmYuZVmdux3gkxULDi14bOBc,32856
+simba/plotting/roi_plotter_mp.py,sha256=vFSDqFjGvqMkG7SbkFJdjDLsNS1TtOVcz5hzYHT5_Tg,32887
 simba/plotting/sam_plotter.py,sha256=XAQuFv-GJkTOno-PeEemUaz0CL83QH1SygJ6Cf1UsgI,6507
 simba/plotting/shap_agg_stats_visualizer.py,sha256=_SqN6vi60JvjTnr2FpxfDdAnGHwuad_ad4TxKQ6tRBY,15614
 simba/plotting/single_run_model_validation_video.py,sha256=75tP0f6UFK-ZpGDWxqLfaWlLrDybkBlEBoN_eKuRTBg,25973
 simba/plotting/single_run_model_validation_video_mp.py,sha256=Mzheon7Mqn_tG11mGcGO_HxHTE_VbMOBmTW1B3pDp_8,27507
 simba/plotting/spontaneous_alternation_plotter.py,sha256=HIgUEMhPCLq7qNXowbFp0bgCPLTvvwf8_DwlOiuWqA8,13117
-simba/plotting/yolo_pose_track_visualizer.py,sha256=uNQIHOPY-mEqgWY6ufkloedZGIgRCB9vyueG2bErVBc,17328
-simba/plotting/yolo_pose_visualizer.py,sha256=_iYELRtBNs6xJze-fyORdoqvdTtswtTvWYVZCX5sDMs,25510
+simba/plotting/yolo_pose_track_visualizer.py,sha256=N01gWdKFD87eHQs20qvgs65CsqjiwyN3Et2qiP_wU2s,17364
+simba/plotting/yolo_pose_visualizer.py,sha256=1K-LpyI40M_5GJ40akL7necAkX56o3AmWN2JeNdeHfU,25543
 simba/plotting/yolo_seg_visualizer.py,sha256=JoGvmESNgqJI9nxId-GVt1mIGoF12Jx3seclz6lxmnA,9221
 simba/plotting/yolo_visualize.py,sha256=6HqMzkboENti6vKrjAzOm19YBrkLFZVbH3SLy-xgBNw,6043
 simba/plotting/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1355,7 +1355,7 @@ simba/ui/import_videos_frame.py,sha256=i0LnQzPLFne9dA_nvpVWAHYGmi0aTRXpiHzEog_-R
 simba/ui/machine_model_settings_ui.py,sha256=hTfpBxtfYGH9Rsf3JdQ5Sc8z874tYAoZefvjF1JD6gA,38292
 simba/ui/ml_settings_frm.py,sha256=f1-E6pEGjWJVF3I6n0azO9zAnsskpZjInViguHIDntw,3101
 simba/ui/px_to_mm_ui.py,sha256=ETedZPFkloU0l3JeGnhiSIAsGBQzzv0YrDWtiVGOmJ4,9387
-simba/ui/tkinter_functions.py,sha256=p880LwVcmnJgtA9Ya_iQdTBsP7p3dVLfF-zCQ7VnHnM,41222
+simba/ui/tkinter_functions.py,sha256=u5QXx6iwlRj3UAsuwg2pC567l4-8IXX0cKppWRCT8P0,41392
 simba/ui/user_defined_pose_creator.py,sha256=QAfdp8r225DONLaCgQJe6kXiZZLj20VQCLJEnCvZsKs,9249
 simba/ui/video_info_ui.py,sha256=ld_fN-3vCYJDFv1t9i5B-VVtoaBCspDmiDUIfOSJb_8,17262
 simba/ui/pop_ups/.DS_Store,sha256=4PUcfpTZzBMd8yz2YahGab5mqPUVIpvdxzjvQ2MmsBY,6148
@@ -1407,7 +1407,7 @@ simba/ui/pop_ups/helpers.py,sha256=wjxintf8wqKfm8SqfDDLVqDGlwKkWbXVtF-e5HwauVs,6
 simba/ui/pop_ups/import_roi_csv_popup.py,sha256=k2kL1ZonBe9J70oUligTGFaSwIW0C8OmdNBr4SjilmA,4795
 simba/ui/pop_ups/initialize_blob_tracking_pop_up.py,sha256=8D0SV2pkkNyzTxi9h5h1rcF4ndRLiwe8pQ7EkZTemG0,2400
 simba/ui/pop_ups/interpolate_pop_up.py,sha256=fz360nYiX6YyFDY3VK7quMyx4w_gzVtygaXVc3xey9g,6284
-simba/ui/pop_ups/kleinberg_pop_up.py,sha256=r_QL1t5IYnH5cVsoVUWtISOFbXKX020g1twUmgtTRVc,4860
+simba/ui/pop_ups/kleinberg_pop_up.py,sha256=aMTWc5OevqKmkpB09NtS3Yxr8R3pNKeFMtLiT1w1PeQ,6196
 simba/ui/pop_ups/labelme_bbox_to_yolo_bbox_popup.py,sha256=iL3g7V5f_Cg_m_D-QA3yPHLRpjMo9vXCPIBwvU4iKn0,3681
 simba/ui/pop_ups/labelme_dir_to_csv_popup.py,sha256=zsObRf09LWCg0X2Wwi7bvfhQT5bXYvfkmUvlIS6k8rM,3611
 simba/ui/pop_ups/labelme_to_df_popup.py,sha256=KE93Dvc-6X44CxnvRFAL5dFkv_i_HC7gwbtaXSZLFhw,4293
@@ -1500,8 +1500,8 @@ simba/utils/data.py,sha256=Zc3BrVGddEgZkm7yvVaBMot8HBz12d_nhHaXhrWB0UM,100827
 simba/utils/enums.py,sha256=ZR2175N06ZHZNiHk8n757T-WGyt1-55LLpcg3Sbb91k,38668
 simba/utils/errors.py,sha256=GeEF9K_9EWHQqZP-ptTM8ABmb9o0SCnec52t0Qw5Tno,18237
 simba/utils/lookups.py,sha256=MTzHIHJpiOtFSTVNQCGMyfY_ItEaurc69QrtCQIfRh8,48918
-simba/utils/printing.py,sha256=L13f5prOnuswDqa4XttEBoQ3y3wNLm0W_ZvyxYQTbYw,5230
-simba/utils/read_write.py,sha256=lKLG-MEgXRtN7NWaLlOa356rUFgdNrWbgtUHrISPa_w,187302
+simba/utils/printing.py,sha256=2s-uESy1knuPiniqQ-q277uQ2teYM4OHo9Y4L20JQWM,5353
+simba/utils/read_write.py,sha256=2786DQVZ88EpYPglLlvDZr9Q_kt-ILcMaJ1zV9hTfIw,187576
 simba/utils/warnings.py,sha256=K7w1RiDL4Un7rGaabOVCGc9fHcaKxk66iZyNLS_AtOE,8121
 simba/utils/yolo.py,sha256=4RTC1JYnZ7WgaWmf70cZATbUtjRQfJ1HyhJhglpAEcs,18482
 simba/utils/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1524,9 +1524,9 @@ simba/video_processors/roi_selector_circle.py,sha256=SD_lv6V3MGiIQd0VtUFSKe83ySW
 simba/video_processors/roi_selector_polygon.py,sha256=DMtilt__gGwNu6VV73CWbnPqrPBXkan1_akUqGEzfGw,6742
 simba/video_processors/video_processing.py,sha256=ho9bz1rjl1eEomA4yfndIc46xtnvdgFNv5JJSX28n7w,311986
 simba/video_processors/videos_to_frames.py,sha256=WqAjEa5enrrkKqwDkL8HhFlbgt1Sf67k2zeCQhVKryM,7323
-simba_uw_tf_dev-4.6.3.dist-info/LICENSE,sha256=Sjn362upcvYFypam-b-ziOXU1Wl5GGuTt5ICrGimzyA,1720
-simba_uw_tf_dev-4.6.3.dist-info/METADATA,sha256=PHM_iOtDP2yiUr1qmqFxpmnqxbV1oVWM3D3ukGU7LwE,11375
-simba_uw_tf_dev-4.6.3.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-simba_uw_tf_dev-4.6.3.dist-info/entry_points.txt,sha256=Nfh_EbfDGdKftLjCnGWtQrBHENiDYMdgupwLyLpU5dc,44
-simba_uw_tf_dev-4.6.3.dist-info/top_level.txt,sha256=ogtimvlqDxDTOBAPfT2WaQ2pGAAbKRXG8z8eUTzf6TU,14
-simba_uw_tf_dev-4.6.3.dist-info/RECORD,,
+simba_uw_tf_dev-4.6.4.dist-info/LICENSE,sha256=Sjn362upcvYFypam-b-ziOXU1Wl5GGuTt5ICrGimzyA,1720
+simba_uw_tf_dev-4.6.4.dist-info/METADATA,sha256=r3UPwczfEURMUuMeXr0Gmk10MY5Hf35_Rx61KgLcVAQ,11375
+simba_uw_tf_dev-4.6.4.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+simba_uw_tf_dev-4.6.4.dist-info/entry_points.txt,sha256=Nfh_EbfDGdKftLjCnGWtQrBHENiDYMdgupwLyLpU5dc,44
+simba_uw_tf_dev-4.6.4.dist-info/top_level.txt,sha256=ogtimvlqDxDTOBAPfT2WaQ2pGAAbKRXG8z8eUTzf6TU,14
+simba_uw_tf_dev-4.6.4.dist-info/RECORD,,