removed band dim form aux_data dataset

awi-response · Oct 11, 2024 · 67cb082 · 67cb082
1 parent c121655
commit 67cb082
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/darts-preprocessing/src/darts_preprocessing/utils/data_pre_processing.py b/darts-preprocessing/src/darts_preprocessing/utils/data_pre_processing.py
@@ -196,7 +196,7 @@ def load_auxiliary(
     # delete temporarary file
     os.remove(outfile)
 
-    return data_array.assign_attrs(ds_annotation).drop_vars("band").to_dataset(name=xr_dataset_name)
+    return data_array.assign_attrs(ds_annotation).drop_vars("band").to_dataset(name=xr_dataset_name).drop_dims("band")
 
 
 def load_data_masks(planet_scene_path):