accusleepy 0.10.0__py3-none-any.whl → 0.10.1__py3-none-any.whl

accusleepy/__init__.py

@@ -0,0 +1 @@
+"""AccuSleePy: Python implementation of AccuSleep for automated sleep scoring."""

accusleepy/__main__.py

@@ -1,3 +1,5 @@
+"""Entry point for running AccuSleePy as a module (python -m accusleepy)."""
+
 from accusleepy.gui.main import run_primary_window
 
 if __name__ == "__main__":

accusleepy/bouts.py

@@ -1,3 +1,5 @@
+"""Brain state bout length enforcement."""
+
 import re
 from dataclasses import dataclass
 from operator import attrgetter

accusleepy/brain_state_set.py

@@ -1,3 +1,5 @@
+"""Define and organize brain states."""
+
 from dataclasses import dataclass
 
 import numpy as np

accusleepy/classification.py

@@ -1,3 +1,5 @@
+"""Model training and brain state classification."""
+
 import os
 
 import numpy as np

accusleepy/constants.py

@@ -1,3 +1,5 @@
+"""Application-wide constants and default settings."""
+
 import numpy as np
 
 # probably don't change these unless you really need to

accusleepy/fileio.py

@@ -1,3 +1,5 @@
+"""File I/O for recordings, labels, calibration data, and configuration."""
+
 import json
 import os
 from dataclasses import dataclass
@@ -5,7 +7,6 @@ from importlib.metadata import version, PackageNotFoundError
 
 import numpy as np
 import pandas as pd
-from PySide6.QtWidgets import QListWidgetItem
 
 from accusleepy.brain_state_set import BRAIN_STATES_KEY, BrainState, BrainStateSet
 import accusleepy.constants as c
@@ -55,7 +56,6 @@ class Recording:
     label_file: str = ""  # path to label file
     calibration_file: str = ""  # path to calibration file
     sampling_rate: int | float = 0.0  # sampling rate, in Hz
-    widget: QListWidgetItem = None  # list item widget shown in the GUI
 
 
 def load_calibration_file(filename: str) -> tuple[np.ndarray, np.ndarray]:

accusleepy/gui/__init__.py

@@ -0,0 +1 @@
+"""Graphical user interface components for AccuSleePy."""

accusleepy/gui/main.py

@@ -1,5 +1,7 @@
-# AccuSleePy main window
-# Icon source: Arkinasi, https://www.flaticon.com/authors/arkinasi
+"""AccuSleePy main window.
+
+Icon source: Arkinasi, https://www.flaticon.com/authors/arkinasi
+"""
 
 import logging
 import os

accusleepy/gui/manual_scoring.py

@@ -1,8 +1,9 @@
-# AccuSleePy manual scoring GUI
-# Icon sources:
-#   Arkinasi, https://www.flaticon.com/authors/arkinasi
-#   kendis lasman, https://www.flaticon.com/packs/ui-79
+"""AccuSleePy manual scoring GUI.
 
+Icon sources:
+    Arkinasi, https://www.flaticon.com/authors/arkinasi
+    kendis lasman, https://www.flaticon.com/packs/ui-79
+"""
 
 import copy
 import os
@@ -85,8 +86,8 @@ BRIGHTER_SCALE_FACTOR = 0.96
 DIMMER_SCALE_FACTOR = 1.07
 # zoom factor for upper plots - larger values = bigger changes
 ZOOM_FACTOR = 0.1
-# interval in seconds between zoom events triggered by scrolling
-ZOOM_DELAY = 0.05
+# rate limit for zoom events triggered by scrolling
+MAX_SCROLL_EVENTS_PER_SEC = 24
 
 
 @dataclass
@@ -983,11 +984,15 @@ class ManualScoringWindow(QDialog):
             return
 
         self.now_zooming = True
+        start_time = time.time()
         if event.button == "up":
             self.zoom_x(direction=ZOOM_IN)
         else:
             self.zoom_x(direction=ZOOM_OUT)
-        time.sleep(ZOOM_DELAY)
+        end_time = time.time()
+        time_elapsed = end_time - start_time
+        if time_elapsed < 1 / MAX_SCROLL_EVENTS_PER_SEC:
+            time.sleep(1 / MAX_SCROLL_EVENTS_PER_SEC - time_elapsed)
         self.now_zooming = False
 
 

accusleepy/gui/mplwidget.py

@@ -1,4 +1,5 @@
-# Widget with a matplotlib FigureCanvas for manual scoring
+"""Matplotlib FigureCanvas widget for manual scoring."""
+
 from collections.abc import Callable
 
 import matplotlib.ticker as mticker

accusleepy/gui/recording_manager.py

@@ -1,11 +1,20 @@
 """Recording list manager"""
 
+from dataclasses import dataclass
+
 from PySide6.QtCore import QObject
 from PySide6.QtWidgets import QListWidget, QListWidgetItem
 
 from accusleepy.fileio import Recording, load_recording_list, save_recording_list
 
 
+@dataclass
+class RecordingListItem(Recording):
+    """A Recording with an associated QListWidget item for the GUI"""
+
+    widget: QListWidgetItem = None
+
+
 class RecordingListManager(QObject):
     """Manages the list of recordings and the associated QListWidget"""
 
@@ -14,19 +23,19 @@ class RecordingListManager(QObject):
         self._widget = list_widget
 
         # Create initial empty recording (there is always at least one)
-        first_recording = Recording(
+        first_recording = RecordingListItem(
             widget=QListWidgetItem("Recording 1", self._widget),
         )
-        self._recordings: list[Recording] = [first_recording]
+        self._recordings: list[RecordingListItem] = [first_recording]
         self._widget.addItem(first_recording.widget)
         self._widget.setCurrentRow(0)
 
     @property
-    def current(self) -> Recording:
+    def current(self) -> RecordingListItem:
         """The currently selected recording"""
         return self._recordings[self._widget.currentRow()]
 
-    def add(self, sampling_rate: int | float) -> Recording:
+    def add(self, sampling_rate: int | float) -> RecordingListItem:
         """Add a new recording to the list
 
         :param sampling_rate: sampling rate for the new recording
@@ -35,7 +44,7 @@ class RecordingListManager(QObject):
         new_name = max(r.name for r in self._recordings) + 1
 
         # Create recording with widget
-        recording = Recording(
+        recording = RecordingListItem(
             name=new_name,
             sampling_rate=sampling_rate,
             widget=QListWidgetItem(f"Recording {new_name}", self._widget),
@@ -63,7 +72,7 @@ class RecordingListManager(QObject):
         else:
             # Reset the single recording to defaults
             recording_name = self._recordings[0].name
-            self._recordings[0] = Recording(widget=self._recordings[0].widget)
+            self._recordings[0] = RecordingListItem(widget=self._recordings[0].widget)
             self._recordings[0].widget.setText(f"Recording {self._recordings[0].name}")
             return f"cleared Recording {recording_name}"
 
@@ -85,14 +94,15 @@ class RecordingListManager(QObject):
         try:
             self._widget.clear()
 
-            # Load recordings
-            self._recordings = load_recording_list(filename)
-
-            # Create widgets for each recording
-            for recording in self._recordings:
-                recording.widget = QListWidgetItem(
-                    f"Recording {recording.name}", self._widget
+            # Load recordings and create widgets
+            self._recordings = [
+                RecordingListItem(
+                    **r.__dict__,
+                    widget=QListWidgetItem(f"Recording {r.name}", self._widget),
                 )
+                for r in load_recording_list(filename)
+            ]
+            for recording in self._recordings:
                 self._widget.addItem(recording.widget)
         finally:
             self._widget.blockSignals(False)
@@ -106,5 +116,5 @@ class RecordingListManager(QObject):
     def __len__(self):
         return len(self._recordings)
 
-    def __getitem__(self, index: int) -> Recording:
+    def __getitem__(self, index: int) -> RecordingListItem:
         return self._recordings[index]

accusleepy/models.py

@@ -1,3 +1,5 @@
+"""Neural network model definitions for sleep stage classification."""
+
 from torch import device, flatten, nn
 from torch import load as torch_load
 from torch import save as torch_save

accusleepy/multitaper.py

@@ -14,10 +14,6 @@ import timeit
 import warnings
 
 import numpy as np
-from joblib import Parallel, cpu_count, delayed
-
-# from scipy.signal import detrend # unused by AccuSleePy
-# from scipy.signal.windows import dpss # lazily loaded later
 
 
 # MULTITAPER SPECTROGRAM #
@@ -206,19 +202,9 @@ def spectrogram(
         wt,
     )
 
-    if multiprocess:  # use multiprocessing
-        n_jobs = max(cpu_count() - 1, 1) if n_jobs is None else n_jobs
-        mt_spectrogram = np.vstack(
-            Parallel(n_jobs=n_jobs)(
-                delayed(calc_mts_segment)(data_segments[num_window, :], *mts_params)
-                for num_window in range(num_windows)
-            )
-        )
-
-    else:  # if no multiprocessing, compute normally
-        mt_spectrogram = np.apply_along_axis(
-            calc_mts_segment, 1, data_segments, *mts_params
-        )
+    mt_spectrogram = np.apply_along_axis(
+        calc_mts_segment, 1, data_segments, *mts_params
+    )
 
     # Compute one-sided PSD spectrum
     mt_spectrogram = mt_spectrogram.T
@@ -251,37 +237,6 @@ def spectrogram(
     if np.all(mt_spectrogram.flatten() == 0):
         print("\n Data was all zeros, no output")
 
-    # # Plot multitaper spectrogram
-    # if plot_on:
-    #     # convert from power to dB
-    #     spect_data = nanpow2db(mt_spectrogram)
-    #
-    #     # Set x and y axes
-    #     dx = stimes[1] - stimes[0]
-    #     dy = sfreqs[1] - sfreqs[0]
-    #     extent = [stimes[0] - dx, stimes[-1] + dx, sfreqs[-1] + dy, sfreqs[0] - dy]
-    #
-    #     # Plot spectrogram
-    #     if ax is None:
-    #         fig, ax = plt.subplots()
-    #     else:
-    #         fig = ax.get_figure()
-    #     im = ax.imshow(spect_data, extent=extent, aspect="auto")
-    #     fig.colorbar(im, ax=ax, label="PSD (dB)", shrink=0.8)
-    #     ax.set_xlabel("Time (HH:MM:SS)")
-    #     ax.set_ylabel("Frequency (Hz)")
-    #     im.set_cmap(plt.cm.get_cmap("cet_rainbow4"))
-    #     ax.invert_yaxis()
-    #
-    #     # Scale colormap
-    #     if clim_scale:
-    #         clim = np.percentile(spect_data, [5, 98])  # from 5th percentile to 98th
-    #         im.set_clim(clim)  # actually change colorbar scale
-    #
-    #     fig.show()
-    #     if return_fig:
-    #         return mt_spectrogram, stimes, sfreqs, (fig, ax)
-
     return mt_spectrogram, stimes, sfreqs
 
 
@@ -568,18 +523,6 @@ def nanpow2db(y):
     return ydB
 
 
-# Helper #
-def is_outlier(data):
-    smad = 1.4826 * np.median(
-        abs(data - np.median(data))
-    )  # scaled median absolute deviation
-    outlier_mask = (
-        abs(data - np.median(data)) > 3 * smad
-    )  # outliers are more than 3 smads away from median
-    outlier_mask = outlier_mask | np.isnan(data) | np.isinf(data)
-    return outlier_mask
-
-
 # CALCULATE MULTITAPER SPECTRUM ON SINGLE SEGMENT
 def calc_mts_segment(
     data_segment,

accusleepy/signal_processing.py

@@ -1,3 +1,5 @@
+"""EEG/EMG signal processing, mixture z-scoring, and training image generation."""
+
 import logging
 import os
 import warnings

accusleepy/temperature_scaling.py

@@ -1,3 +1,5 @@
+"""Temperature scaling for model confidence calibration."""
+
 import logging
 
 import numpy as np

accusleepy/validation.py

@@ -1,3 +1,5 @@
+"""Validation utilities for brain state labels and recordings."""
+
 import numpy as np
 
 from accusleepy.brain_state_set import BrainStateSet

{accusleepy-0.10.0.dist-info → accusleepy-0.10.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: accusleepy
-Version: 0.10.0
+Version: 0.10.1
 Summary: Python implementation of AccuSleep
 License: GPL-3.0-only
 Author: Zeke Barger
@@ -11,14 +11,12 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: fastparquet (>=2024.11.0,<2025.0.0)
-Requires-Dist: joblib (>=1.4.2,<2.0.0)
 Requires-Dist: matplotlib (>=3.10.1,<4.0.0)
 Requires-Dist: numpy (>=2.2.4,<3.0.0)
 Requires-Dist: pandas (>=2.2.3,<3.0.0)
 Requires-Dist: pillow (>=11.1.0,<12.0.0)
-Requires-Dist: pre-commit (>=4.2.0,<5.0.0)
-Requires-Dist: pyside6 (>=6.9.0,<6.9.3)
+Requires-Dist: pyarrow (>=23.0.0,<24.0.0)
+Requires-Dist: pyside6 (>=6.10.1,<7.0.0)
 Requires-Dist: scipy (>=1.15.2,<2.0.0)
 Requires-Dist: torch (>=2.8.0,<3.0.0)
 Requires-Dist: torchvision (>=0.23.0,<1.0.0)
@@ -79,10 +77,8 @@ please consult the [developer guide](accusleepy/gui/text/dev_guide.md).
 
 ## Changelog
 
-- 0.10.0: Improved zoom behavior
-- 0.8.1-0.9.3: Improved error handling and code quality
-- 0.8.0: More configurable settings, visual improvements
-- 0.7.1-0.7.3: Bugfixes, code cleanup
+- 0.10.0-0.10.1: Improved zoom behavior, updated dependencies
+- 0.7.1-0.9.3: Bugfixes, code cleanup, additional config settings
 - 0.7.0: More settings can be configured in the UI
 - 0.6.0: Confidence scores can now be displayed and saved. Retraining your models is recommended
     since the new calibration feature will make the confidence scores more accurate.

{accusleepy-0.10.0.dist-info → accusleepy-0.10.1.dist-info}/RECORD

@@ -1,12 +1,12 @@
-accusleepy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-accusleepy/__main__.py,sha256=dKzl2N2Hg9lD264CWYNxThRyDKzWwyMwHRXmJxOmMis,104
-accusleepy/bouts.py,sha256=GVYOSdsQjHzYirRRSFlie8w5iJq74umyGNMRVsQlpK0,6045
-accusleepy/brain_state_set.py,sha256=i9RUwSMKG21AjromjOJYpsHkip3elTFhHYntMguBF6Q,3273
-accusleepy/classification.py,sha256=UJ48QamTvqwqBwu-r4weSno-9TQyPFCk62nbUc5_iKA,8672
+accusleepy/__init__.py,sha256=MeKOECkuTv2gZTnlCQsI1qbMNNaAfNKBNJ2vkG87KhU,82
+accusleepy/__main__.py,sha256=yRSJnSLFjt9BqzmdemfMzejp1TPS0VGIAvyYcd-yg3w,182
+accusleepy/bouts.py,sha256=s77xtimYWFJ9CjTzqMUu98A2odoT8waoyuE6FVmj1dA,6089
+accusleepy/brain_state_set.py,sha256=la7CjgwMByXEIE__Yzv-G9mbuNYFIB6G41rDPP2T21k,3314
+accusleepy/classification.py,sha256=Zh-iMjvttRWf-Oarr1urgT-FkjcCbzYgndooAdeZbTA,8726
 accusleepy/config.json,sha256=vtzWKXSwZRNlYQhyaIl-TvOPrOxgRC7Pt3fUcWChVeo,911
-accusleepy/constants.py,sha256=DjfUeGiiL6NTfw0Rgh37QGcsoCHVVK1PVC3l7P-JUis,3225
-accusleepy/fileio.py,sha256=W_3UomA-MC6SPJqvcxcT5d9rHm1sVY090t0-lE4Aeuw,9324
-accusleepy/gui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+accusleepy/constants.py,sha256=zB8z5-PWpnfZMROkR7iZusqP3cxRyr5sVBbGSCxAAto,3281
+accusleepy/fileio.py,sha256=50_eC6rJ98cEdspJfhQSyOWoTNX0Q9FG5ZvU2e6Y9Ls,9283
+accusleepy/gui/__init__.py,sha256=bv2FO3aQKzcBifoEfzg3ZeZHQaHUY_SuEceIE5mzZ7k,58
 accusleepy/gui/dialogs.py,sha256=DqkarJKgRovTuDDJhqPaw_o8ZfopuR7R6M8C49QWZu0,1230
 accusleepy/gui/icons/brightness_down.png,sha256=PLT1fb83RHIhSRuU7MMMx0G7oJAY7o9wUcnqM8veZfM,12432
 accusleepy/gui/icons/brightness_up.png,sha256=64GnUqgPvN5xZ6Um3wOzwqvUmdAWYZT6eFmWpBsHyks,12989
@@ -19,15 +19,15 @@ accusleepy/gui/icons/save.png,sha256=J3EA8iU1BqLYRSsrq_OdoZlqrv2yfL7oV54DklTy_DI
 accusleepy/gui/icons/up_arrow.png,sha256=V9yF9t1WgjPaUu-mF1YGe_DfaRHg2dUpR_sUVVcvVvY,3329
 accusleepy/gui/icons/zoom_in.png,sha256=MFWnKZp7Rvh4bLPq4Cqo4sB_jQYedUUtT8-ZO8tNYyc,13589
 accusleepy/gui/icons/zoom_out.png,sha256=IB8Jecb3i0U4qjWRR46ridjLpvLCSe7PozBaLqQqYSw,13055
-accusleepy/gui/images/primary_window.png,sha256=8zREhWe4Xbk_gRkeaHdgC7jyYzjUBGyDj8CUVHLB9c4,568383
+accusleepy/gui/images/primary_window.png,sha256=1L-82Zl2adVpfRuFQKecYwz9G_Dthcoyq-XvmFXQbrg,568426
 accusleepy/gui/images/viewer_window.png,sha256=b_B7m9WSLMAOzNjctq76SyekO1WfC6qYZVNnYfhjPe8,977197
 accusleepy/gui/images/viewer_window_annotated.png,sha256=uMNUmsZIdzDlQpyoiS3lJGoWlg_T325Oj5hDZhM3Y14,146817
-accusleepy/gui/main.py,sha256=VT2Nz8shk1Z3riEtq8j0tkYCs3RTJAC9_j3HXhh2eOg,24934
-accusleepy/gui/manual_scoring.py,sha256=85dy6hRT4Pz5nhUwd2w6UB4KR2lOVJdHUmOoD2At8O8,43157
-accusleepy/gui/mplwidget.py,sha256=rJSTtWmLjHn8r3c9Kb23Rc4XzXl3i9B-JrjNjjlNnmQ,13492
+accusleepy/gui/main.py,sha256=CqVlUw7ICrxNUaz97tPNwoLW7fhJoAORyHsH43mzKS8,24939
+accusleepy/gui/manual_scoring.py,sha256=F_o6rh9tXLMv21s8zWJXu7YZKUkQ-GvsXRBvOGPBx0I,43365
+accusleepy/gui/mplwidget.py,sha256=FRH3A86L_b7cZR6BVY-fvqzJ6iFh-KdNYIwJQbMZPP4,13491
 accusleepy/gui/primary_window.py,sha256=QwYpzE47AU-Pt6Ftvyvjivz2U9QgElTWrOLq8jXG5co,141860
 accusleepy/gui/primary_window.ui,sha256=iy24sjevVI3Kr_lt7t1fjazlA-wE1dq94jsOBU6Oloc,207947
-accusleepy/gui/recording_manager.py,sha256=0lYrGu-7gX0xpLxBBcn4-TcuSy8CsmiPBJ3l_fZa158,3916
+accusleepy/gui/recording_manager.py,sha256=CqgP8Q2lI7Vir-zAOhVlZbIVBxRAbgC6ydmvzlMLYz4,4207
 accusleepy/gui/resources.qrc,sha256=wqPendnTLAuKfVI6v2lKHiRqAWM0oaz2ZuF5cucJdS4,803
 accusleepy/gui/resources_rc.py,sha256=Z2e34h30U4snJjnYdZVV9B6yjATKxxfvgTRt5uXtQdo,329727
 accusleepy/gui/settings_widget.py,sha256=9KIqFg7s-a7VyED-lNxS2Zux1hTVJLy6AdSlw2expD8,17489
@@ -36,12 +36,12 @@ accusleepy/gui/text/main_guide.md,sha256=XDJU2anRsA72zXOSkhstSy66uEIA4jitD4uCHIx
 accusleepy/gui/text/manual_scoring_guide.md,sha256=mzIqYJ5IOkzIpAWNZUh82ve_lI9SHjVvbQzn5kDWP_k,1065
 accusleepy/gui/viewer_window.py,sha256=jysFw7C_Tr7mtK1XNWhIpHblBvatwduE3RF2GP4lrro,24479
 accusleepy/gui/viewer_window.ui,sha256=a89iVLk1sJg9N6ZvWAV6YNPStb2Tm4-rs-W7TmIDkb4,31658
-accusleepy/models.py,sha256=MHMsC9sYfR3fTOQLh4COyjCZo9q2vaX1FmpTHaaVtM4,3516
-accusleepy/multitaper.py,sha256=D5-iglwkFBRciL5tKSNcunMtcq0rM3zHwRHUVPgem1U,25679
+accusleepy/models.py,sha256=Kmbt6a8kIbPD8fIPVHIo60oywQNNH86_lj1dSsjPkFE,3588
+accusleepy/multitaper.py,sha256=X5ZGNfbYVmsX4EPsv1IqpiFaCZYR0KQruyMGUUomJfk,23623
 accusleepy/services.py,sha256=4xl9uWDaJXIpJTYH08mSgipkkEYM9fDLIcMj7-k9qf0,20228
-accusleepy/signal_processing.py,sha256=k3a2IKchok3-38P7T0XHNpGTiWKBqWtQm7SHwFEkTNA,19973
-accusleepy/temperature_scaling.py,sha256=RXsbDLT5JDrjPbz5ZdFXKj6NxpQCHpq5BzcLBzsUJKc,5848
-accusleepy/validation.py,sha256=h3iHVW71AaUC4lp1KMQXo5j81bS-IuYbTZWRSIkmHWE,6933
-accusleepy-0.10.0.dist-info/METADATA,sha256=KX6di9ZBZgVy_18KAaxKEOnNOLlQr9tVY1IdtOKciPM,4728
-accusleepy-0.10.0.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
-accusleepy-0.10.0.dist-info/RECORD,,
+accusleepy/signal_processing.py,sha256=-itwidKi7Gy9rukSezJ_1IJ9XFI_D-0j4IpUZaFnTKE,20057
+accusleepy/temperature_scaling.py,sha256=PE-xzFeZ0vdLJKmaJ1mjDUeqsb-pSp_y9BsaINXHFas,5909
+accusleepy/validation.py,sha256=4cR70x0yombDA5EWISmXRpbGY5YLsjzaLzDBBa3wnUc,7000
+accusleepy-0.10.1.dist-info/METADATA,sha256=UEVePOEL8dZbVfvJa5-zAtDhMLOVfG15GACvEpilAN4,4582
+accusleepy-0.10.1.dist-info/WHEEL,sha256=kJCRJT_g0adfAJzTx2GUMmS80rTJIVHRCfG0DQgLq3o,88
+accusleepy-0.10.1.dist-info/RECORD,,

{accusleepy-0.10.0.dist-info → accusleepy-0.10.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.3.0
+Generator: poetry-core 2.3.1
 Root-Is-Purelib: true
 Tag: py3-none-any