pylance standardization

src/analysis/participantbrain.py

@@ -1,18 +1,28 @@
 """
 Filename: participantbrain.py
 Description: Logic for the Participant Brain analysis window
+Note: Compliant with pylance strict type checking
 
 Author: Tyler de Zeeuw
 License: GPL-3.0
 """
 
+# Built-in imports
+from pathlib import Path
+from typing import Any, cast
+
 # External library imports
+from mne import Annotations
+from pandas import DataFrame
+
+from mne.io.base import BaseRaw
+
 from flares import brain_3d_visualization, brain_landmarks_3d
 from src.shared.flaresbasewidget import ParticipantUIMixin, FlaresBaseWidget
 from src.shared.shareddata import APP_NAME
 
 
-PARAMETERIZED_INDEXES = {
+PARAMETERIZED_INDEXES: dict[int, list[dict[str, Any]]] = {
     0: [
         {
             "key": "show_optodes",
@@ -57,7 +67,12 @@ PARAMETERIZED_INDEXES = {
 
 
 class ParticipantBrainViewerWidget(ParticipantUIMixin, FlaresBaseWidget):
-    def __init__(self, haemo_dict, cha_dict):
+    def __init__(
+        self, 
+        haemo_dict: dict[str | Path, BaseRaw], 
+        cha_dict: dict[str, DataFrame],
+    ) -> None:
+        
         super().__init__("ParticipantBrain")
         self.setWindowTitle(f"Participant Brain Viewer - {APP_NAME.upper()}")
         self.haemo_dict = haemo_dict
@@ -72,21 +87,31 @@ class ParticipantBrainViewerWidget(ParticipantUIMixin, FlaresBaseWidget):
         if request is None:
             return
 
-        (selected_event, selected_file_paths, selected_indexes, param_values,) = request  
+        (selected_event, selected_file_paths, selected_indexes, raw_params) = request  
+
+        param_values = cast(dict[int | str, dict[str, Any]], raw_params)
 
         # Pass the necessary arguments to each method
         for file_path in selected_file_paths:
             haemo_obj = self.haemo_dict.get(file_path)
 
+            if haemo_obj is None:
+                continue
+
             if selected_event:
-                participant_events = set(haemo_obj.annotations.description)
+                raw_annotations = getattr(haemo_obj, "annotations", None)
+                
+                if raw_annotations is not None:
+                    annotations = cast(Annotations, raw_annotations)
+                    descriptions = cast(list[str], list(annotations.description))
+                    participant_events: set[str] = set(descriptions)
+                else:
+                    participant_events: set[str] = set()
+                    
                 if selected_event not in participant_events:
                     print(f"Skipping {self.participant_map[file_path]}: Event '{selected_event}' not found.")
                     continue
 
-            if haemo_obj is None:
-                raise Exception("How did we get here?")
-
             cha = self.cha_dict.get(file_path)
 
             for idx in selected_indexes: