2d-arrary-support

src/ria_toolkit_oss/viz/radio_dataset.py

@@ -88,7 +88,10 @@ def _check_dataset_compatibility(dataset, plot_type: str) -> tuple[bool, str]:
             # Check if we can access sample data (basic test)
             try:
                 sample_data = dataset[0] if hasattr(dataset, "__getitem__") else None
-                if sample_data is None or len(sample_data) < 32:
+                if sample_data is None:
+                    return False, "Insufficient sample data for spectrogram (need at least 32 points)"
+                n_pts = sample_data.shape[-1] if hasattr(sample_data, "shape") and sample_data.ndim > 1 else len(sample_data)
+                if n_pts < 32:
                     return False, "Insufficient sample data for spectrogram (need at least 32 points)"
             except Exception:
                 # If we can't access data, we'll rely on synthetic data generation