boris-behav-obs 8.12__py3-none-any.whl → 9.7.6__py3-none-any.whl

boris/time_budget_widget.py

@@ -1,7 +1,7 @@
 """
 BORIS
 Behavioral Observation Research Interactive Software
-Copyright 2012-2023 Olivier Friard
+Copyright 2012-2025 Olivier Friard
 
 This file is part of BORIS.
 
@@ -26,6 +26,7 @@ import pathlib as pl
 from decimal import Decimal as dec
 from io import StringIO
 import pandas as pd
+import time
 
 try:
     import pyreadr
@@ -36,8 +37,8 @@ except ModuleNotFoundError:
 
 
 import tablib
-from PyQt5.QtCore import Qt
-from PyQt5.QtWidgets import (
+from PySide6.QtCore import Qt
+from PySide6.QtWidgets import (
     QFileDialog,
     QHBoxLayout,
     QInputDialog,
@@ -51,6 +52,7 @@ from PyQt5.QtWidgets import (
     QTableWidgetItem,
     QVBoxLayout,
     QWidget,
+    QApplication,
 )
 
 from . import config as cfg
@@ -110,7 +112,7 @@ class timeBudgetResults(QWidget):
         spacerItem = QSpacerItem(241, 20, QSizePolicy.Expanding, QSizePolicy.Minimum)
         hbox2.addItem(spacerItem)
 
-        self.pbClose = QPushButton("Close", clicked=self.close_clicked)
+        self.pbClose = QPushButton(cfg.CLOSE, clicked=self.close_clicked)
         hbox2.addWidget(self.pbClose)
 
         hbox.addLayout(hbox2)
@@ -129,48 +131,47 @@ class timeBudgetResults(QWidget):
         save time budget analysis results in TSV, CSV, ODS, XLS format
         """
 
-        def complete(l: list, max_: int) -> list:
+        def complete(lst: list, max_: int) -> list:
             """
             complete list with empty string until len = max
 
             Args:
-                l (list): list to complete
+                lst (list): list to complete
                 max_ (int): length of the returned list
 
             Returns:
                 list: completed list
             """
 
-            l.extend([""] * (max_ - len(l)))
-            return l
+            lst.extend([""] * (max_ - len(lst)))
+            return lst
 
         logging.debug("save time budget results to file")
 
-        file_formats = [cfg.TSV, cfg.CSV, cfg.ODS, cfg.XLSX, cfg.XLS, cfg.HTML, cfg.PANDAS_DF, cfg.RDS]
+        file_formats = (cfg.TSV, cfg.CSV, cfg.ODS, cfg.XLSX, cfg.XLS, cfg.HTML, cfg.TEXT_FILE, cfg.PANDAS_DF, cfg.RDS)
 
-        file_name, filter_ = QFileDialog().getSaveFileName(
-            self, "Save Time budget analysis", "", ";;".join(file_formats)
-        )
+        file_name, filter_ = QFileDialog().getSaveFileName(self, "Save Time budget analysis", "", ";;".join(file_formats))
 
         if not file_name:
             return
 
         # add correct file extension if not present
         if pl.Path(file_name).suffix != f".{cfg.FILE_NAME_SUFFIX[filter_]}":
-            file_name = str(pl.Path(file_name)) + "." + cfg.FILE_NAME_SUFFIX[filter_]
+            if cfg.FILE_NAME_SUFFIX[filter_] != "cli":
+                file_name = str(pl.Path(file_name)) + "." + cfg.FILE_NAME_SUFFIX[filter_]
+            else:
+                file_name = str(pl.Path(file_name))
             # check if file with new extension already exists
             if pl.Path(file_name).is_file():
                 if (
-                    dialog.MessageDialog(
-                        cfg.programName, f"The file {file_name} already exists.", [cfg.CANCEL, cfg.OVERWRITE]
-                    )
+                    dialog.MessageDialog(cfg.programName, f"The file {file_name} already exists.", [cfg.CANCEL, cfg.OVERWRITE])
                     == cfg.CANCEL
                 ):
                     return
 
-        rows = []
+        rows: list = []
 
-        header = ["Observation id", "Observation date", "Description"]
+        header: list = ["Observation id", "Observation date", "Description"]
         # indep var labels
         header.extend([self.pj[cfg.INDEPENDENT_VARIABLES][idx]["label"] for idx in self.pj[cfg.INDEPENDENT_VARIABLES]])
         header.extend(["Time budget start", "Time budget stop", "Time budget duration"])
@@ -203,9 +204,9 @@ class timeBudgetResults(QWidget):
         for idx in self.pj.get(cfg.INDEPENDENT_VARIABLES, []):
             if self.lw.count() == 1:
                 # var has value in obs?
-                if self.pj[cfg.INDEPENDENT_VARIABLES][idx]["label"] in self.pj[cfg.OBSERVATIONS][
-                    self.lw.item(0).text()
-                ].get(cfg.INDEPENDENT_VARIABLES, []):
+                if self.pj[cfg.INDEPENDENT_VARIABLES][idx]["label"] in self.pj[cfg.OBSERVATIONS][self.lw.item(0).text()].get(
+                    cfg.INDEPENDENT_VARIABLES, []
+                ):
                     col1.append(
                         self.pj[cfg.OBSERVATIONS][self.lw.item(0).text()][cfg.INDEPENDENT_VARIABLES][
                             self.pj[cfg.INDEPENDENT_VARIABLES][idx]["label"]
@@ -221,10 +222,10 @@ class timeBudgetResults(QWidget):
             col1.extend([f"{self.min_time:0.3f}", f"{self.max_time:0.3f}", f"{self.max_time - self.min_time:0.3f}"])
 
         if self.time_interval == cfg.TIME_FULL_OBS:
-            col1.extend([f"Full observation", f"Full observation", f"Full observation"])
+            col1.extend(["Full observation", "Full observation", "Full observation"])
 
         if self.time_interval == cfg.TIME_EVENTS:
-            col1.extend([f"Limited to coded events", f"Limited to coded events", f"Limited to coded events"])
+            col1.extend(["Limited to coded events", "Limited to coded events", "Limited to coded events"])
 
         for row_idx in range(self.twTB.rowCount()):
             values = []
@@ -376,7 +377,7 @@ class timeBudgetResults(QWidget):
 
         # write results
         with open(file_name, "wb") as f:
-            if filter_ in (cfg.TSV, cfg.CSV, cfg.HTML):
+            if filter_ in (cfg.TSV, cfg.CSV, cfg.HTML, cfg.TEXT_FILE):
                 f.write(str.encode(data.export(cfg.FILE_NAME_SUFFIX[filter_])))
             if filter_ in (cfg.ODS, cfg.XLSX, cfg.XLS):
                 f.write(data.export(cfg.FILE_NAME_SUFFIX[filter_]))
@@ -389,6 +390,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
     Args:
         mode (str): ["by_behavior", "by_category"]
         mode2 (str): must be in ["list", "current"]
+                     "current" time budget of current observation
     """
 
     if mode2 == "current":
@@ -415,9 +417,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
     flagGroup: bool = False
     if len(selected_observations) > 1:
         flagGroup = (
-            dialog.MessageDialog(
-                cfg.programName, "Group the selected observations in a single time budget analysis?", [cfg.YES, cfg.NO]
-            )
+            dialog.MessageDialog(cfg.programName, "Group the selected observations in a single time budget analysis?", [cfg.YES, cfg.NO])
             == cfg.YES
         )
 
@@ -425,20 +425,25 @@ def time_budget(self, mode: str, mode2: str = "list"):
         self.pj[cfg.OBSERVATIONS], selected_observations
     )
 
-    logging.debug(
-        f"max_media_duration_all_obs: {max_media_duration_all_obs}, total_media_duration_all_obs={total_media_duration_all_obs}"
-    )
+    logging.debug(f"max_media_duration_all_obs: {max_media_duration_all_obs}, total_media_duration_all_obs={total_media_duration_all_obs}")
 
     start_coding, end_coding, _ = observation_operations.coding_time(self.pj[cfg.OBSERVATIONS], selected_observations)
 
-    parameters = select_subj_behav.choose_obs_subj_behav_category(
+    start_interval, end_interval = observation_operations.time_intervals_range(self.pj[cfg.OBSERVATIONS], selected_observations)
+
+    parameters: dict = select_subj_behav.choose_obs_subj_behav_category(
         self,
         selected_observations,
         start_coding=start_coding,
         end_coding=end_coding,
+        # start_interval=start_interval,
+        # end_interval=end_interval,
+        start_interval=None,
+        end_interval=None,
         maxTime=max_media_duration_all_obs,
         by_category=(mode == "by_category"),
         n_observations=len(selected_observations),
+        show_exclude_non_coded_modifiers=True,
     )
     if parameters == {}:
         return
@@ -447,21 +452,27 @@ def time_budget(self, mode: str, mode2: str = "list"):
         QMessageBox.warning(None, cfg.programName, "Select subject(s) and behavior(s) to analyze")
         return
 
+    logging.debug(f"{parameters=}")
+
     # ask for excluding behaviors durations from total time
-    if not start_coding.is_nan():
+    if start_coding is not None and not start_coding.is_nan():
         cancel_pressed, parameters[cfg.EXCLUDED_BEHAVIORS] = self.filter_behaviors(
             title="Select behaviors to exclude from the total time",
-            text=("The duration of the selected behaviors will " "be subtracted from the total time"),
+            text=("The duration of the selected behaviors will be subtracted from the total time"),
             table="",
-            behavior_type=[cfg.STATE_EVENT],
+            behavior_type=cfg.STATE_EVENT_TYPES,
         )
         if cancel_pressed:
             return
     else:
         parameters[cfg.EXCLUDED_BEHAVIORS] = []
 
+    self.statusbar.showMessage(f"Generating time budget for {len(selected_observations)} observation(s)")
+    QApplication.processEvents()
+
     # check if time_budget window must be used
     if flagGroup or len(selected_observations) == 1:
+        t0 = time.time()
 
         cursor = db_functions.load_events_in_db(
             self.pj,
@@ -473,7 +484,6 @@ def time_budget(self, mode: str, mode2: str = "list"):
 
         total_observation_time = 0
         for obsId in selected_observations:
-
             obs_length = observation_operations.observation_total_length(self.pj[cfg.OBSERVATIONS][obsId])
 
             if obs_length == dec(-1):  # media length not available
@@ -482,7 +492,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
             if obs_length == dec(-2):  # images obs without time
                 parameters[cfg.TIME_INTERVAL] = cfg.TIME_EVENTS
 
-            if parameters[cfg.TIME_INTERVAL] == cfg.TIME_FULL_OBS:
+            if parameters[cfg.TIME_INTERVAL] == cfg.TIME_FULL_OBS:  # media file duration
                 min_time = float(0)
                 # check if the last event is recorded after media file length
                 try:
@@ -493,7 +503,14 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 except Exception:
                     max_time = float(obs_length)
 
-            if parameters[cfg.TIME_INTERVAL] == cfg.TIME_EVENTS:
+            if parameters[cfg.TIME_INTERVAL] == cfg.TIME_OBS_INTERVAL:
+                obs_interval = self.pj[cfg.OBSERVATIONS][obsId].get(cfg.OBSERVATION_TIME_INTERVAL, [0, 0])
+                offset = float(self.pj[cfg.OBSERVATIONS][obsId][cfg.TIME_OFFSET])
+                min_time = float(obs_interval[0]) + offset
+                # Use max media duration for max time if no interval is defined (=0)
+                max_time = float(obs_interval[1]) + offset if obs_interval[1] != 0 else float(obs_length)
+
+            if parameters[cfg.TIME_INTERVAL] == cfg.TIME_EVENTS:  # events duration
                 try:
                     min_time = float(self.pj[cfg.OBSERVATIONS][obsId][cfg.EVENTS][0][0])  # first event
                 except Exception:
@@ -511,8 +528,10 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 # check intervals
                 for subj in parameters[cfg.SELECTED_SUBJECTS]:
                     for behav in parameters[cfg.SELECTED_BEHAVIORS]:
-                        if cfg.POINT in self.eventType(behav).upper():
+                        # if cfg.POINT in self.eventType(behav).upper():
+                        if project_functions.event_type(behav, self.pj[cfg.ETHOGRAM]) in cfg.POINT_EVENT_TYPES:
                             continue
+
                         # extract modifiers
 
                         cursor.execute(
@@ -524,7 +543,6 @@ def time_budget(self, mode: str, mode2: str = "list"):
                         # logging.debug("distinct_modifiers: {}".format(distinct_modifiers))
 
                         for modifier in distinct_modifiers:
-
                             # logging.debug("modifier #{}#".format(modifier[0]))
 
                             # insert events at boundaries of time interval
@@ -541,12 +559,8 @@ def time_budget(self, mode: str, mode2: str = "list"):
                                 )
                                 % 2
                             ):
-
                                 cursor.execute(
-                                    (
-                                        "INSERT INTO events (observation, subject, code, type, modifiers, occurence) "
-                                        "VALUES (?,?,?,?,?,?)"
-                                    ),
+                                    ("INSERT INTO events (observation, subject, code, type, modifiers, occurence) VALUES (?,?,?,?,?,?)"),
                                     (obsId, subj, behav, "STATE", modifier[0], min_time),
                                 )
 
@@ -562,12 +576,8 @@ def time_budget(self, mode: str, mode2: str = "list"):
                                 )
                                 % 2
                             ):
-
                                 cursor.execute(
-                                    (
-                                        "INSERT INTO events (observation, subject, code, type, modifiers, occurence) "
-                                        "VALUES (?,?,?,?,?,?)"
-                                    ),
+                                    ("INSERT INTO events (observation, subject, code, type, modifiers, occurence) VALUES (?,?,?,?,?,?)"),
                                     (obsId, subj, behav, "STATE", modifier[0], max_time),
                                 )
                         try:
@@ -582,6 +592,10 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 "DELETE FROM events WHERE observation = ? AND (occurence < ? OR occurence > ?)",
                 (obsId, min_time, max_time),
             )
+            try:
+                cursor.execute("COMMIT")
+            except Exception:
+                pass
 
         out, categories = time_budget_functions.time_budget_analysis(
             self.pj[cfg.ETHOGRAM], cursor, selected_observations, parameters, by_category=(mode == "by_category")
@@ -593,9 +607,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
             if element["subject"] not in excl_behaviors_total_time:
                 excl_behaviors_total_time[element["subject"]] = 0
             if element["behavior"] in parameters[cfg.EXCLUDED_BEHAVIORS]:
-                excl_behaviors_total_time[element["subject"]] += (
-                    element["duration"] if not isinstance(element["duration"], str) else 0
-                )
+                excl_behaviors_total_time[element["subject"]] += element["duration"] if not isinstance(element["duration"], str) else 0
 
         # widget for results visualization
         self.tb = timeBudgetResults(self.pj, self.config_param)
@@ -615,20 +627,14 @@ def time_budget(self, mode: str, mode2: str = "list"):
         if len(selected_observations) > 1:
             if total_observation_time:
                 if self.timeFormat == cfg.HHMMSS:
-                    self.tb.lbTotalObservedTime.setText(
-                        f"Total observation length: {util.seconds2time(total_observation_time)}"
-                    )
+                    self.tb.lbTotalObservedTime.setText(f"Total observation length: {util.seconds2time(total_observation_time)}")
                 if self.timeFormat == cfg.S:
-                    self.tb.lbTotalObservedTime.setText(
-                        f"Total observation length: {float(total_observation_time):0.3f}"
-                    )
+                    self.tb.lbTotalObservedTime.setText(f"Total observation length: {float(total_observation_time):0.3f}")
             else:
                 self.tb.lbTotalObservedTime.setText("Total observation length: not available")
         else:
             if self.timeFormat == cfg.HHMMSS:
-                self.tb.lbTotalObservedTime.setText(
-                    f"Analysis from {util.seconds2time(min_time)} to {util.seconds2time(max_time)}"
-                )
+                self.tb.lbTotalObservedTime.setText(f"Analysis from {util.seconds2time(min_time)} to {util.seconds2time(max_time)}")
             if self.timeFormat == cfg.S:
                 self.tb.lbTotalObservedTime.setText(f"Analysis from {float(min_time):0.3f} to {float(max_time):0.3f} s")
 
@@ -640,8 +646,10 @@ def time_budget(self, mode: str, mode2: str = "list"):
         else:
             self.tb.excluded_behaviors_list.setVisible(False)
 
-        if mode == "by_behavior":
+        self.statusbar.showMessage(f"Time budget generated in {round(time.time() - t0, 3)} s", 5000)
+        logging.debug("Time budget generated")
 
+        if mode == "by_behavior":
             tb_fields = [
                 "Subject",
                 "Behavior",
@@ -673,7 +681,6 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 self.tb.twTB.setRowCount(self.tb.twTB.rowCount() + 1)
                 column = 0
                 for field in fields:
-
                     if isinstance(row[field], float):
                         item = QTableWidgetItem(f"{row[field]:.3f}")
                     else:
@@ -689,10 +696,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 elif row["duration"] not in ("-", cfg.UNPAIRED) and not start_coding.is_nan():
                     tot_time = float(total_observation_time)
                     # substract time of excluded behaviors from the total for the subject
-                    if (
-                        row["subject"] in excl_behaviors_total_time
-                        and row["behavior"] not in parameters[cfg.EXCLUDED_BEHAVIORS]
-                    ):
+                    if row["subject"] in excl_behaviors_total_time and row["behavior"] not in parameters[cfg.EXCLUDED_BEHAVIORS]:
                         tot_time -= excl_behaviors_total_time[row["subject"]]
                     item = QTableWidgetItem(f"{row['duration'] / tot_time * 100:.1f}" if tot_time > 0 else cfg.NA)
 
@@ -703,7 +707,6 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 self.tb.twTB.setItem(self.tb.twTB.rowCount() - 1, column, item)
 
         if mode == "by_category":
-
             tb_fields = ["Subject", "Category", "Total number", "Total duration (s)"]
             fields = ["number", "duration"]
 
@@ -711,9 +714,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
             self.tb.twTB.setHorizontalHeaderLabels(tb_fields)
 
             for subject in categories:
-
                 for category in categories[subject]:
-
                     self.tb.twTB.setRowCount(self.tb.twTB.rowCount() + 1)
 
                     column = 0
@@ -745,12 +746,11 @@ def time_budget(self, mode: str, mode2: str = "list"):
 
         self.tb.twTB.resizeColumnsToContents()
 
-        gui_utilities.restore_geometry(self.tb, "time budget", (0, 0))
+        gui_utilities.restore_geometry(self.tb, "time budget", (800, 600))
 
         self.tb.show()
 
     if not flagGroup and len(selected_observations) > 1:
-
         output_format, ok = QInputDialog.getItem(
             self,
             "Time budget analysis format",
@@ -776,9 +776,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
         if output_format in (cfg.ODS_WB, cfg.XLSX_WB):
             workbook = tablib.Databook()
 
-            wb_file_name, filter_ = QFileDialog(self).getSaveFileName(
-                self, "Save Time budget analysis", "", output_format
-            )
+            wb_file_name, filter_ = QFileDialog(self).getSaveFileName(self, "Save Time budget analysis", "", output_format)
             if not wb_file_name:
                 return
 
@@ -787,9 +785,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 # check if file with new extension already exists
                 if pl.Path(wb_file_name).is_file():
                     if (
-                        dialog.MessageDialog(
-                            cfg.programName, f"The file {wb_file_name} already exists.", [cfg.CANCEL, cfg.OVERWRITE]
-                        )
+                        dialog.MessageDialog(cfg.programName, f"The file {wb_file_name} already exists.", [cfg.CANCEL, cfg.OVERWRITE])
                         == cfg.CANCEL
                     ):
                         return
@@ -805,7 +801,6 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 return
 
         if mode == "by_behavior":
-
             tb_fields = [
                 "Subject",
                 "Behavior",
@@ -831,23 +826,19 @@ def time_budget(self, mode: str, mode2: str = "list"):
             ]
 
         if mode == "by_category":
-
             tb_fields = ["Subject", "Category", "Total number of occurences", "Total duration (s)"]
             fields = ["subject", "category", "number", "duration"]
 
         mem_command = ""
         for obsId in selected_observations:
-
-            cursor = db_functions.load_events_in_db(
-                self.pj, parameters[cfg.SELECTED_SUBJECTS], [obsId], parameters[cfg.SELECTED_BEHAVIORS]
-            )
+            cursor = db_functions.load_events_in_db(self.pj, parameters[cfg.SELECTED_SUBJECTS], [obsId], parameters[cfg.SELECTED_BEHAVIORS])
 
             obs_length = observation_operations.observation_total_length(self.pj[cfg.OBSERVATIONS][obsId])
 
             if obs_length == -1:
                 obs_length = 0
 
-            if parameters["time"] == cfg.TIME_FULL_OBS:
+            if parameters[cfg.TIME_INTERVAL] == cfg.TIME_FULL_OBS:
                 min_time = float(0)
                 # check if the last event is recorded after media file length
                 try:
@@ -858,7 +849,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 except Exception:
                     max_time = float(obs_length)
 
-            if parameters["time"] == cfg.TIME_EVENTS:
+            if parameters[cfg.TIME_INTERVAL] == cfg.TIME_EVENTS:
                 try:
                     min_time = float(self.pj[cfg.OBSERVATIONS][obsId][cfg.EVENTS][0][0])
                 except Exception:
@@ -868,19 +859,15 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 except Exception:
                     max_time = float(obs_length)
 
-            if parameters["time"] == cfg.TIME_ARBITRARY_INTERVAL:
+            if parameters[cfg.TIME_INTERVAL] == cfg.TIME_ARBITRARY_INTERVAL:
                 min_time = float(parameters[cfg.START_TIME])
                 max_time = float(parameters[cfg.END_TIME])
 
                 # check intervals
                 for subj in parameters[cfg.SELECTED_SUBJECTS]:
                     for behav in parameters[cfg.SELECTED_BEHAVIORS]:
-                        if cfg.POINT in project_functions.event_type(
-                            behav, self.pj[cfg.ETHOGRAM]
-                        ):  # self.eventType(behav).upper():
+                        if project_functions.event_type(behav, self.pj[cfg.ETHOGRAM]) in cfg.POINT_EVENT_TYPES:
                             continue
-                        # extract modifiers
-                        # if plot_parameters["include modifiers"]:
 
                         cursor.execute(
                             "SELECT distinct modifiers FROM events WHERE observation = ? AND subject = ? AND code = ?",
@@ -889,7 +876,6 @@ def time_budget(self, mode: str, mode2: str = "list"):
                         distinct_modifiers = list(cursor.fetchall())
 
                         for modifier in distinct_modifiers:
-
                             if (
                                 len(
                                     cursor.execute(
@@ -904,10 +890,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
                                 % 2
                             ):
                                 cursor.execute(
-                                    (
-                                        "INSERT INTO events (observation, subject, code, type, modifiers, occurence) "
-                                        "VALUES (?,?,?,?,?,?)"
-                                    ),
+                                    ("INSERT INTO events (observation, subject, code, type, modifiers, occurence) VALUES (?,?,?,?,?,?)"),
                                     (obsId, subj, behav, "STATE", modifier[0], min_time),
                                 )
                             if (
@@ -923,10 +906,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
                                 % 2
                             ):
                                 cursor.execute(
-                                    (
-                                        "INSERT INTO events (observation, subject, code, type, modifiers, occurence) "
-                                        "VALUES (?,?,?,?,?,?)"
-                                    ),
+                                    ("INSERT INTO events (observation, subject, code, type, modifiers, occurence) VALUES (?,?,?,?,?,?)"),
                                     (obsId, subj, behav, cfg.STATE, modifier[0], max_time),
                                 )
                         try:
@@ -949,9 +929,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 if element["subject"] not in excl_behaviors_total_time:
                     excl_behaviors_total_time[element["subject"]] = 0
                 if element["behavior"] in parameters[cfg.EXCLUDED_BEHAVIORS]:
-                    excl_behaviors_total_time[element["subject"]] += (
-                        element["duration"] if element["duration"] != "NA" else 0
-                    )
+                    excl_behaviors_total_time[element["subject"]] += element["duration"] if element["duration"] != "NA" else 0
 
             rows: list = []
             col1: list = []
@@ -965,9 +943,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
             indep_var_values: list = []
             for _, v in self.pj.get(cfg.INDEPENDENT_VARIABLES, {}).items():
                 indep_var_label.append(v["label"])
-                indep_var_values.append(
-                    self.pj[cfg.OBSERVATIONS][obsId].get(cfg.INDEPENDENT_VARIABLES, {}).get(v["label"], "")
-                )
+                indep_var_values.append(self.pj[cfg.OBSERVATIONS][obsId].get(cfg.INDEPENDENT_VARIABLES, {}).get(v["label"], ""))
 
             header.extend(indep_var_label)
             col1.extend(indep_var_values)
@@ -980,7 +956,6 @@ def time_budget(self, mode: str, mode2: str = "list"):
             header.extend(["Time budget start", "Time budget stop", "Time budget duration"])
 
             if mode == "by_behavior":
-
                 # header
                 rows.append(header + tb_fields)
 
@@ -994,10 +969,7 @@ def time_budget(self, mode: str, mode2: str = "list"):
                     elif row["duration"] not in ("-", cfg.UNPAIRED) and not start_coding.is_nan():
                         tot_time = float(max_time - min_time)
                         # substract duration of excluded behaviors from total time for each subject
-                        if (
-                            row["subject"] in excl_behaviors_total_time
-                            and row["behavior"] not in parameters[cfg.EXCLUDED_BEHAVIORS]
-                        ):
+                        if row["subject"] in excl_behaviors_total_time and row["behavior"] not in parameters[cfg.EXCLUDED_BEHAVIORS]:
                             tot_time -= excl_behaviors_total_time[row["subject"]]
                         # % of tot time
                         values.append(round(row["duration"] / tot_time * 100, 1) if tot_time > 0 else cfg.NA)
@@ -1010,7 +982,6 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 rows.append(header + tb_fields)
 
                 for subject in categories:
-
                     for category in categories[subject]:
                         values = []
                         values.append(subject)
@@ -1039,7 +1010,6 @@ def time_budget(self, mode: str, mode2: str = "list"):
                 workbook.add_sheet(data)
 
             else:
-
                 file_name = f"{pl.Path(exportDir) / pl.Path(util.safeFileName(obsId))}.{extension}"
                 if mem_command != cfg.OVERWRITE_ALL and pl.Path(file_name).is_file():
                     if mem_command == "Skip all":