Changing the band array for testing purposes on databricks

OCHA-DAP · Nov 7, 2024 · 00d097e · 00d097e
1 parent 53b6076
commit 00d097e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/utils/raster_utils.py b/src/utils/raster_utils.py
@@ -283,9 +283,9 @@ def upsample_raster(ds, resampled_resolution=UPSAMPLED_RESOLUTION, logger=None):
                 # Falls under different bands, use the long_name instead of integer value
                 #ds_ = ds_.expand_dims({'band': [ds_.long_name[val-1]]})
                 if int(ds_['band']) ==1 :
-                    ds_ = ds_.expand_dims({'band': 'SFED'})
+                    ds_ = ds_.expand_dims({'band': ['SFED']})
                 else:
-                    ds_ = ds_.expand_dims({'band': 'MFED'})
+                    ds_ = ds_.expand_dims({'band': ['MFED']})
             else:
                 ds_ = ds_.expand_dims([fourth_dim])
             resampled_arrays.append(ds_)