pylance standardization

src/analysis/crossgroupbrainimage.py

@@ -1,20 +1,28 @@
 """
 Filename: crossgroupbrainimage.py
 Description: Logic for the Cross-Group Brain & Image 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.io.base import BaseRaw
+
 import pandas as pd
+from pandas import DataFrame
 
 from flares import aggregate_fnirs_group_geometry, plot_2d_3d_contrasts_between_groups
 from src.shared.flaresbasewidget import CrossGroupUIMixin, FlaresBaseWidget
 from src.shared.shareddata import APP_NAME
 
 
-PARAMETERIZED_INDEXES = {
+PARAMETERIZED_INDEXES: dict[int, list[dict[str, Any]]] = {
     0: [
         {
             "key": "show_optodes",
@@ -52,7 +60,15 @@ PARAMETERIZED_INDEXES = {
 
 
 class CrossGroupBrainImageWidget(CrossGroupUIMixin, FlaresBaseWidget):
-    def __init__(self, haemo_dict, df_ind_dict, design_matrix_dict, contrast_results_dict, group_dict):
+    def __init__(
+        self,
+        haemo_dict: dict[str | Path, BaseRaw], 
+        df_ind_dict: dict[str, DataFrame],
+        design_matrix_dict: dict[str, DataFrame],
+        contrast_results_dict: dict[str, dict[str, Any]],
+        group_dict: dict[str, str],
+    ) -> None:
+        
         super().__init__("CrossGroupBrainImage")
         self.setWindowTitle(f"Cross-Group Brain & Image Viewer - {APP_NAME.upper()}")
         self.haemo_dict = haemo_dict
@@ -70,8 +86,9 @@ class CrossGroupBrainImageWidget(CrossGroupUIMixin, FlaresBaseWidget):
         if request is None:
             return
 
-        (selected_event, file_paths_a, file_paths_b, all_selected_paths, selected_indexes, param_values,) = request
+        (selected_event, file_paths_a, file_paths_b, all_selected_paths, selected_indexes, raw_params) = request
 
+        param_values = cast(dict[int | str, dict[str, Any]], raw_params)
 
         # Build group-level contrast DataFrames
         def concat_group_contrasts(file_paths: list[str], event: str | None) -> pd.DataFrame:
@@ -102,8 +119,11 @@ class CrossGroupBrainImageWidget(CrossGroupUIMixin, FlaresBaseWidget):
         
         if len(all_raw_objs) > 1:
             processed_raw = aggregate_fnirs_group_geometry(all_raw_objs)
+        elif len(all_raw_objs) == 1 and all_raw_objs[0] is not None:
+            processed_raw = all_raw_objs[0].copy()
+            processed_raw.pick(picks="hbo")  # type: ignore
         else:
-            processed_raw = all_raw_objs[0].copy().pick(picks="hbo")
+            processed_raw = None
 
         # Visualizations
         for idx in selected_indexes: