boris-behav-obs 8.16.5__py3-none-any.whl → 9.7.1__py3-none-any.whl

boris/plot_events.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.
 
@@ -25,8 +25,10 @@ import pathlib as pl
 
 import matplotlib
 
-matplotlib.use("Qt5Agg")
+matplotlib.use("QtAgg")
+
 import matplotlib.dates
+
 import matplotlib.pyplot as plt
 import numpy as np
 from matplotlib.dates import (
@@ -37,15 +39,15 @@ from . import config as cfg
 from . import db_functions, project_functions, observation_operations
 from . import utilities as util
 
+# matplotlib.pyplot.switch_backend("Qt5Agg")
+
 
 def default_value(ethogram, behavior, parameter):
     """
     return value for duration in case of point event
     """
     default_value_ = 0
-    if project_functions.event_type(
-        behavior, ethogram
-    ) == "POINT EVENT" and parameter in ["duration"]:
+    if project_functions.event_type(behavior, ethogram) in cfg.POINT_EVENT_TYPES and parameter in ["duration"]:
         default_value_ = "NA"
     return default_value_
 
@@ -65,7 +67,6 @@ def init_behav_modif(
     for subj in selected_subjects:
         behaviors[subj] = {}
         for behav_modif in distinct_behav_modif:
-
             behav, modif = behav_modif
             behav_modif_str = "|".join(behav_modif) if modif else behav
 
@@ -73,16 +74,12 @@ def init_behav_modif(
                 behaviors[subj][behav_modif_str] = {}
 
             for parameter in parameters:
-                behaviors[subj][behav_modif_str][parameter[0]] = default_value(
-                    ethogram, behav_modif_str, parameter[0]
-                )
+                behaviors[subj][behav_modif_str][parameter[0]] = default_value(ethogram, behav_modif_str, parameter[0])
 
     return behaviors
 
 
-def init_behav(
-    ethogram: dict, selected_subjects: list, distinct_behaviors, parameters
-) -> dict:
+def init_behav(ethogram: dict, selected_subjects: list, distinct_behaviors, parameters) -> dict:
     """
     initialize dictionary with subject, behaviors and modifiers
     """
@@ -94,9 +91,7 @@ def init_behav(
             if behavior not in behaviors[subj]:
                 behaviors[subj][behavior] = {}
             for parameter in parameters:
-                behaviors[subj][behavior][parameter] = default_value(
-                    ethogram, behavior, parameter
-                )
+                behaviors[subj][behavior][parameter] = default_value(ethogram, behavior, parameter)
     return behaviors
 
 
@@ -128,9 +123,7 @@ def create_behaviors_bar_plot(
 
     parameters = ["duration", "number of occurences"]
 
-    ok, msg, db_connector = db_functions.load_aggregated_events_in_db(
-        pj, selected_subjects, selected_observations, selected_behaviors
-    )
+    ok, msg, db_connector = db_functions.load_aggregated_events_in_db(pj, selected_subjects, selected_observations, selected_behaviors)
 
     if not ok:
         return {"error": True, "message": msg}
@@ -139,7 +132,6 @@ def create_behaviors_bar_plot(
     all_behaviors = util.all_behaviors(pj[cfg.ETHOGRAM])
 
     for obs_id in selected_observations:
-
         cursor = db_connector.cursor()
         # distinct behaviors
         cursor.execute(
@@ -163,16 +155,12 @@ def create_behaviors_bar_plot(
         )
         distinct_subjects = [rows["subject"] for rows in cursor.fetchall()]
 
-        behaviors = init_behav(
-            pj[cfg.ETHOGRAM], distinct_subjects, distinct_behav, parameters
-        )
+        behaviors = init_behav(pj[cfg.ETHOGRAM], distinct_subjects, distinct_behav, parameters)
 
         # plot creation
         if len(distinct_subjects) > 1:
             fig, axs = plt.subplots(nrows=1, ncols=len(distinct_subjects), sharey=True)
-            fig2, axs2 = plt.subplots(
-                nrows=1, ncols=len(distinct_subjects), sharey=True
-            )
+            fig2, axs2 = plt.subplots(nrows=1, ncols=len(distinct_subjects), sharey=True)
 
         else:
             fig, ax = plt.subplots(nrows=1, ncols=len(distinct_subjects), sharey=True)
@@ -190,9 +178,7 @@ def create_behaviors_bar_plot(
         cursor.execute("UPDATE aggregated_events SET modifiers = ''")
 
         # time
-        obs_length = observation_operations.observation_total_length(
-            pj[cfg.OBSERVATIONS][obs_id]
-        )
+        obs_length = observation_operations.observation_total_length(pj[cfg.OBSERVATIONS][obs_id])
         if obs_length == -1:
             obs_length = 0
 
@@ -210,7 +196,7 @@ def create_behaviors_bar_plot(
             except Exception:
                 max_time = float(obs_length)
 
-        if param["time"] == cfg.TIME_ARBITRARY_INTERVAL:
+        if param["time"] in (cfg.TIME_ARBITRARY_INTERVAL, cfg.TIME_OBS_INTERVAL):
             min_time = float(start_time)
             max_time = float(end_time)
 
@@ -246,15 +232,10 @@ def create_behaviors_bar_plot(
         )
 
         for ax_idx, subject in enumerate(sorted(distinct_subjects)):
-
             for behavior in distinct_behav:
-
                 # number of occurences
                 cursor.execute(
-                    (
-                        "SELECT COUNT(*) AS count FROM aggregated_events "
-                        "WHERE observation = ? AND subject = ? AND behavior = ?"
-                    ),
+                    ("SELECT COUNT(*) AS count FROM aggregated_events WHERE observation = ? AND subject = ? AND behavior = ?"),
                     (
                         obs_id,
                         subject,
@@ -262,14 +243,10 @@ def create_behaviors_bar_plot(
                     ),
                 )
                 for row in cursor.fetchall():
-                    behaviors[subject][behavior]["number of occurences"] = (
-                        0 if row["count"] is None else row["count"]
-                    )
+                    behaviors[subject][behavior]["number of occurences"] = 0 if row["count"] is None else row["count"]
 
                 # total duration
-                if cfg.STATE in project_functions.event_type(
-                    behavior, pj[cfg.ETHOGRAM]
-                ):
+                if project_functions.event_type(behavior, pj[cfg.ETHOGRAM]) in cfg.STATE_EVENT_TYPES:
                     cursor.execute(
                         (
                             "SELECT SUM(stop - start) AS duration FROM aggregated_events "
@@ -282,9 +259,7 @@ def create_behaviors_bar_plot(
                         ),
                     )
                     for row in cursor.fetchall():
-                        behaviors[subject][behavior]["duration"] = (
-                            0 if row["duration"] is None else row["duration"]
-                        )
+                        behaviors[subject][behavior]["duration"] = 0 if row["duration"] is None else row["duration"]
 
             (
                 durations,
@@ -296,24 +271,14 @@ def create_behaviors_bar_plot(
             ) = ([], [], [], [], [], [])
 
             for behavior in sorted(distinct_behav):
-
-                if (
-                    param[cfg.EXCLUDE_BEHAVIORS]
-                    and behaviors[subject][behavior]["number of occurences"] == 0
-                ):
+                if param[cfg.EXCLUDE_BEHAVIORS] and behaviors[subject][behavior]["number of occurences"] == 0:
                     continue
 
-                n_occurences.append(
-                    behaviors[subject][behavior]["number of occurences"]
-                )
+                n_occurences.append(behaviors[subject][behavior]["number of occurences"])
                 x_labels.append(behavior)
 
                 # color
-                behav_idx = [
-                    k
-                    for k in pj[cfg.ETHOGRAM]
-                    if pj[cfg.ETHOGRAM][k]["code"] == behavior
-                ][0]
+                behav_idx = [k for k in pj[cfg.ETHOGRAM] if pj[cfg.ETHOGRAM][k]["code"] == behavior][0]
                 col = None
                 if cfg.COLOR in pj[cfg.ETHOGRAM][behav_idx]:
                     col = util.behavior_user_color(pj[cfg.ETHOGRAM], behavior)
@@ -321,17 +286,11 @@ def create_behaviors_bar_plot(
                     colors.append(col)
                 else:
                     try:
-                        colors.append(
-                            util.behavior_color(
-                                plot_colors, all_behaviors.index(behavior)
-                            )
-                        )
+                        colors.append(util.behavior_color(plot_colors, all_behaviors.index(behavior)))
                     except Exception:
                         colors.append("darkgray")
 
-                if cfg.STATE in project_functions.event_type(
-                    behavior, pj[cfg.ETHOGRAM]
-                ):
+                if project_functions.event_type(behavior, pj[cfg.ETHOGRAM]) in cfg.STATE_EVENT_TYPES:
                     durations.append(behaviors[subject][behavior]["duration"])
                     x_labels_duration.append(behavior)
 
@@ -342,11 +301,7 @@ def create_behaviors_bar_plot(
                         colors_duration.append(col)
                     else:
                         try:
-                            colors_duration.append(
-                                util.behavior_color(
-                                    plot_colors, all_behaviors.index(behavior)
-                                )
-                            )
+                            colors_duration.append(util.behavior_color(plot_colors, all_behaviors.index(behavior)))
                         except Exception:
                             colors_duration.append("darkgray")
 
@@ -372,9 +327,7 @@ def create_behaviors_bar_plot(
             axs[ax_idx].set_title(f"{subject}")
 
             axs[ax_idx].set_xticks(np.arange(len(durations)))
-            axs[ax_idx].set_xticklabels(
-                x_labels_duration, rotation="vertical", fontsize=8
-            )
+            axs[ax_idx].set_xticklabels(x_labels_duration, rotation="vertical", fontsize=8)
 
             if ax_idx == 0:
                 axs2[ax_idx].set_ylabel("Number of occurences")
@@ -392,12 +345,8 @@ def create_behaviors_bar_plot(
 
         if plot_directory:
             # output_file_name = f"{pathlib.Path(plot_directory) / utilities.safeFileName(obs_id)}.{output_format}"
-            fig.savefig(
-                f"{pl.Path(plot_directory) / util.safeFileName(obs_id)}.duration.{output_format}"
-            )
-            fig2.savefig(
-                f"{pl.Path(plot_directory) / util.safeFileName(obs_id)}.number_of_occurences.{output_format}"
-            )
+            fig.savefig(f"{pl.Path(plot_directory) / util.safeFileName(obs_id)}.duration.{output_format}")
+            fig2.savefig(f"{pl.Path(plot_directory) / util.safeFileName(obs_id)}.number_of_occurences.{output_format}")
             plt.close()
         else:
             fig.show()
@@ -426,9 +375,7 @@ def create_events_plot(
     start_time = parameters[cfg.START_TIME]
     end_time = parameters[cfg.END_TIME]
 
-    ok, msg, db_connector = db_functions.load_aggregated_events_in_db(
-        self.pj, selected_subjects, selected_observations, selected_behaviors
-    )
+    ok, msg, db_connector = db_functions.load_aggregated_events_in_db(self.pj, selected_subjects, selected_observations, selected_behaviors)
 
     if not ok:
         return False, msg, None
@@ -440,9 +387,7 @@ def create_events_plot(
         cursor.execute("UPDATE aggregated_events SET modifiers = ''")
 
     cursor.execute("SELECT DISTINCT behavior, modifiers FROM aggregated_events")
-    distinct_behav_modif = [
-        [rows["behavior"], rows["modifiers"]] for rows in cursor.fetchall()
-    ]
+    distinct_behav_modif = [[rows["behavior"], rows["modifiers"]] for rows in cursor.fetchall()]
 
     # add selected behaviors that are not observed
     for behav in selected_behaviors:
@@ -452,31 +397,21 @@ def create_events_plot(
     distinct_behav_modif = sorted(distinct_behav_modif)
     max_len = len(distinct_behav_modif)
 
-    all_behaviors = [
-        self.pj[cfg.ETHOGRAM][x][cfg.BEHAVIOR_CODE]
-        for x in util.sorted_keys(self.pj[cfg.ETHOGRAM])
-    ]
+    all_behaviors = [self.pj[cfg.ETHOGRAM][x][cfg.BEHAVIOR_CODE] for x in util.sorted_keys(self.pj[cfg.ETHOGRAM])]
 
     par1 = 1
     bar_height = 0.5
     epoch_date = dt.datetime(2017, 1, 1)
 
     for obs_id in selected_observations:
-
         if len(selected_subjects) > 1:
-            fig, axs = plt.subplots(
-                figsize=(20, 8), nrows=len(selected_subjects), ncols=1, sharex=True
-            )
+            fig, axs = plt.subplots(figsize=(20, 8), nrows=len(selected_subjects), ncols=1, sharex=True)
         else:
-            fig, ax = plt.subplots(
-                figsize=(20, 8), nrows=len(selected_subjects), ncols=1, sharex=True
-            )
+            fig, ax = plt.subplots(figsize=(20, 8), nrows=len(selected_subjects), ncols=1, sharex=True)
             axs = np.ndarray(shape=(1), dtype=type(ax))
             axs[0] = ax
 
-        ok, msg, db_connector = db_functions.load_aggregated_events_in_db(
-            self.pj, selected_subjects, [obs_id], selected_behaviors
-        )
+        ok, msg, db_connector = db_functions.load_aggregated_events_in_db(self.pj, selected_subjects, [obs_id], selected_behaviors)
 
         cursor = db_connector.cursor()
         # if modifiers not to be included set modifiers to ""
@@ -485,9 +420,7 @@ def create_events_plot(
         cursor = db_connector.cursor()
 
         cursor.execute("SELECT DISTINCT behavior, modifiers FROM aggregated_events")
-        distinct_behav_modif = [
-            [rows["behavior"], rows["modifiers"]] for rows in cursor.fetchall()
-        ]
+        distinct_behav_modif = [[rows["behavior"], rows["modifiers"]] for rows in cursor.fetchall()]
 
         # add selected behaviors that are not observed
         if not parameters["exclude behaviors"]:
@@ -499,9 +432,7 @@ def create_events_plot(
         max_len = len(distinct_behav_modif)
 
         # time
-        obs_length = observation_operations.observation_total_length(
-            self.pj[cfg.OBSERVATIONS][obs_id]
-        )
+        obs_length = observation_operations.observation_total_length(self.pj[cfg.OBSERVATIONS][obs_id])
         if obs_length == -1:  # media length not available
             interval = cfg.TIME_EVENTS
 
@@ -509,17 +440,20 @@ def create_events_plot(
             min_time = 0.0
             max_time = float(obs_length)
 
+        if interval == cfg.TIME_OBS_INTERVAL:
+            obs_interval = self.pj[cfg.OBSERVATIONS][obs_id].get(cfg.OBSERVATION_TIME_INTERVAL, [0, 0])
+            offset = float(self.pj[cfg.OBSERVATIONS][obs_id][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 interval == cfg.TIME_EVENTS:
             try:
-                min_time = float(
-                    self.pj[cfg.OBSERVATIONS][obs_id][cfg.EVENTS][0][0]
-                )  # first event
+                min_time = float(self.pj[cfg.OBSERVATIONS][obs_id][cfg.EVENTS][0][0])  # first event
             except Exception:
                 min_time = 0.0
             try:
-                max_time = float(
-                    self.pj[cfg.OBSERVATIONS][obs_id][cfg.EVENTS][-1][0]
-                )  # last event
+                max_time = float(self.pj[cfg.OBSERVATIONS][obs_id][cfg.EVENTS][-1][0])  # last event
             except Exception:
                 max_time = float(obs_length)
 
@@ -566,15 +500,12 @@ def create_events_plot(
 
         ylabels = [" ".join(x) for x in distinct_behav_modif]
         for ax_idx, subject in enumerate(selected_subjects):
-
             if parameters["exclude behaviors"]:
                 cursor.execute(
                     "SELECT DISTINCT behavior, modifiers FROM aggregated_events WHERE subject = ?",
                     (subject,),
                 )
-                distinct_behav_modif = [
-                    [rows["behavior"], rows["modifiers"]] for rows in cursor.fetchall()
-                ]
+                distinct_behav_modif = [[rows["behavior"], rows["modifiers"]] for rows in cursor.fetchall()]
 
                 # add selected behaviors that are not observed
                 if not parameters["exclude behaviors"]:
@@ -594,17 +525,12 @@ def create_events_plot(
             i = 0
             for behavior_modifiers in distinct_behav_modif:
                 behavior, modifiers = behavior_modifiers
-                behavior_modifiers_str = (
-                    "|".join(behavior_modifiers) if modifiers else behavior
-                )
+                behavior_modifiers_str = "|".join(behavior_modifiers) if modifiers else behavior
                 bars[behavior_modifiers_str] = []
 
                 # total duration
                 cursor.execute(
-                    (
-                        "SELECT start, stop FROM aggregated_events "
-                        "WHERE  subject = ? AND behavior = ? AND modifiers = ?"
-                    ),
+                    ("SELECT start, stop FROM aggregated_events WHERE  subject = ? AND behavior = ? AND modifiers = ?"),
                     (
                         subject,
                         behavior,
@@ -615,27 +541,16 @@ def create_events_plot(
                     bars[behavior_modifiers_str].append((row["start"], row["stop"]))
 
                     if self.timeFormat == cfg.HHMMSS:
-                        start_date = matplotlib.dates.date2num(
-                            epoch_date + dt.timedelta(seconds=row["start"])
-                        )
+                        start_date = matplotlib.dates.date2num(epoch_date + dt.timedelta(seconds=row["start"]))
                         end_date = matplotlib.dates.date2num(
-                            epoch_date
-                            + dt.timedelta(
-                                seconds=row["stop"]
-                                + cfg.POINT_EVENT_PLOT_DURATION
-                                * (row["stop"] == row["start"])
-                            )
+                            epoch_date + dt.timedelta(seconds=row["stop"] + cfg.POINT_EVENT_PLOT_DURATION * (row["stop"] == row["start"]))
                         )
                     if self.timeFormat == cfg.S:
                         start_date = row["start"]
                         end_date = row["stop"]
 
                     # color
-                    behav_idx = [
-                        k
-                        for k in self.pj[cfg.ETHOGRAM]
-                        if self.pj[cfg.ETHOGRAM][k]["code"] == behavior
-                    ][0]
+                    behav_idx = [k for k in self.pj[cfg.ETHOGRAM] if self.pj[cfg.ETHOGRAM][k]["code"] == behavior][0]
                     col = None
                     if cfg.COLOR in self.pj[cfg.ETHOGRAM][behav_idx]:
                         col = util.behavior_user_color(self.pj[cfg.ETHOGRAM], behavior)
@@ -643,16 +558,10 @@ def create_events_plot(
                         bar_color = col
                     else:
                         try:
-                            bar_color = util.behavior_color(
-                                plot_colors, all_behaviors.index(behavior)
-                            )
+                            bar_color = util.behavior_color(plot_colors, all_behaviors.index(behavior))
                         except Exception:
                             bar_color = "darkgray"
-                    bar_color = (
-                        cfg.POINT_EVENT_PLOT_COLOR
-                        if row["stop"] == row["start"]
-                        else bar_color
-                    )
+                    bar_color = cfg.POINT_EVENT_PLOT_COLOR if row["stop"] == row["start"] else bar_color
 
                     # sage colors removed from matplotlib colors list
                     if bar_color in ("sage", "darksage", "lightsage"):
@@ -700,12 +609,8 @@ def create_events_plot(
 
             if self.timeFormat == cfg.HHMMSS:
                 axs[ax_idx].set_xlim(
-                    left=matplotlib.dates.date2num(
-                        epoch_date + dt.timedelta(seconds=min_time)
-                    ),
-                    right=matplotlib.dates.date2num(
-                        epoch_date + dt.timedelta(seconds=max_time)
-                    ),
+                    left=matplotlib.dates.date2num(epoch_date + dt.timedelta(seconds=min_time)),
+                    right=matplotlib.dates.date2num(epoch_date + dt.timedelta(seconds=max_time)),
                 )
 
             axs[ax_idx].grid(color="g", linestyle=":")
@@ -724,8 +629,6 @@ def create_events_plot(
         plt.tight_layout()
 
         if len(selected_observations) > 1:
-            plt.savefig(
-                f"{pl.Path(plot_directory) / util.safeFileName(obs_id)}.{file_format}"
-            )
+            plt.savefig(f"{pl.Path(plot_directory) / util.safeFileName(obs_id)}.{file_format}")
         else:
             plt.show()

boris/plot_events_rt.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.
 
@@ -25,26 +25,27 @@ Plot events in real time
 
 import matplotlib
 
-matplotlib.use("Qt5Agg")
+matplotlib.use("QtAgg")
+
 import numpy as np
-from PyQt5.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QLabel
-from PyQt5.QtCore import pyqtSignal, QEvent
-from PyQt5 import Qt
-from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
+from PySide6.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QLabel
+from PySide6.QtCore import Signal, QEvent, Qt
+from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg as FigureCanvas
 
 from matplotlib.figure import Figure
 
 from . import config as cfg
 
+# matplotlib.pyplot.switch_backend("Qt5Agg")
 
-class Plot_events_RT(QWidget):
 
+class Plot_events_RT(QWidget):
     # send keypress event to mainwindow
-    sendEvent = pyqtSignal(QEvent)
+    sendEvent = Signal(QEvent)
 
     def __init__(self):
         super().__init__()
-        self.setWindowTitle(f"Events plot")
+        self.setWindowTitle("Events plot")
 
         self.interval = 60  # default interval of visualization (in seconds)
         self.time_mem = -1
@@ -65,13 +66,9 @@ class Plot_events_RT(QWidget):
 
         hlayout1 = QHBoxLayout()
         hlayout1.addWidget(QLabel("Time interval"))
-        hlayout1.addWidget(
-            QPushButton("+", self, clicked=lambda: self.time_interval_changed(1), focusPolicy=Qt.Qt.NoFocus)
-        )
-        hlayout1.addWidget(
-            QPushButton("-", self, clicked=lambda: self.time_interval_changed(-1), focusPolicy=Qt.Qt.NoFocus)
-        )
-        self.pb_mode = QPushButton("Include modifiers", self, clicked=self.change_mode, focusPolicy=Qt.Qt.NoFocus)
+        hlayout1.addWidget(QPushButton("+", self, clicked=lambda: self.time_interval_changed(1), focusPolicy=Qt.NoFocus))
+        hlayout1.addWidget(QPushButton("-", self, clicked=lambda: self.time_interval_changed(-1), focusPolicy=Qt.NoFocus))
+        self.pb_mode = QPushButton("Include modifiers", self, clicked=self.change_mode, focusPolicy=Qt.NoFocus)
         hlayout1.addWidget(self.pb_mode)
         layout.addLayout(hlayout1)
 
@@ -147,13 +144,11 @@ class Plot_events_RT(QWidget):
             intervals_behav[group(event[1], event[2], event[3])] = [(0, 0)]
 
         for event in events:
-
             time_, subject, code, modifier = event[:4]
             key = group(subject, code, modifier)
 
             # check if code is state
             if code in self.state_events_list:
-
                 if key in mem_behav and mem_behav[key] is not None:
                     # stop interval
 
@@ -172,22 +167,16 @@ class Plot_events_RT(QWidget):
                     mem_behav[key] = time_
 
             else:  # point event
-
                 if start <= time_ <= end:
-                    intervals_behav[key].append(
-                        (float(time_), float(time_) + self.point_event_plot_duration * 50)
-                    )  # point event -> 1 s
+                    intervals_behav[key].append((float(time_), float(time_) + self.point_event_plot_duration * 50))  # point event -> 1 s
 
         # check if intervals are closed
         for k in mem_behav:
             if mem_behav[k] is not None:  # interval open
                 if self.observation_type == cfg.LIVE:
-                    intervals_behav[k].append(
-                        (float(mem_behav[k]), float((end + start) / 2))
-                    )  # close interval with current time
+                    intervals_behav[k].append((float(mem_behav[k]), float((end + start) / 2)))  # close interval with current time
 
                 elif self.observation_type == cfg.MEDIA:
-
                     intervals_behav[k].append((float(mem_behav[k]), float(end)))  # close interval with end value
 
         return intervals_behav
@@ -201,9 +190,7 @@ class Plot_events_RT(QWidget):
             force_plot (bool): force plot even if media paused
         """
 
-        self.events = self.aggregate_events(
-            self.events_list, current_time - self.interval / 2, current_time + self.interval / 2
-        )
+        self.events = self.aggregate_events(self.events_list, current_time - self.interval / 2, current_time + self.interval / 2)
 
         if not force_plot and current_time == self.time_mem:
             return
@@ -211,7 +198,6 @@ class Plot_events_RT(QWidget):
         self.time_mem = current_time
 
         if self.events != self.events_mem:
-
             left, duration = {}, {}
             for k in self.events:
                 left[k] = []