.

sorghum_pipeline/pipeline.py

@@ -99,7 +99,14 @@ class SorghumPipeline:
             green_band = None
             spectral = pdata.get('spectral_stack', {})
             if 'green' in spectral:
-                green_band = spectral['green'].squeeze(-1).astype(np.float64)
+                gb = spectral['green']
+                gb = np.asarray(gb)
+                # robustly collapse to 2D if it arrived as 3-channel
+                if gb.ndim == 3 and gb.shape[2] > 1:
+                    gb = gb[..., 0]
+                elif gb.ndim == 3 and gb.shape[2] == 1:
+                    gb = gb.squeeze(-1)
+                green_band = gb.astype(np.float64)
                 if mask is not None:
                     valid = np.where(mask > 0, green_band, np.nan)
                 else:
@@ -138,7 +145,14 @@ class SorghumPipeline:
             if not all(b in spectral for b in bands):
                 continue
             
-            arrays = [np.asarray(spectral[b].squeeze(-1), dtype=np.float64) for b in bands]
+            arrays = []
+            for b in bands:
+                arr = np.asarray(spectral[b])
+                if arr.ndim == 3 and arr.shape[2] > 1:
+                    arr = arr[..., 0]
+                elif arr.ndim == 3 and arr.shape[2] == 1:
+                    arr = arr.squeeze(-1)
+                arrays.append(arr.astype(np.float64))
             values = self.vegetation_extractor.index_formulas[name](*arrays).astype(np.float64)
             binary_mask = (mask > 0)
             masked_values = np.where(binary_mask, values, np.nan)