Merge pull request #15 from umr-lops/landmask-management

fix bug with pol attr
umr-lops · Feb 15, 2024 · 6bce46b · 6bce46b
2 parents 73bbec2 + 957b798
commit 6bce46b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/l2awinddirection/mainl2awinddirection.py b/l2awinddirection/mainl2awinddirection.py
@@ -247,7 +247,11 @@ def main():
             res = generate_wind_product(tiles, model_m64rn4)
         res.attrs["winddirection_L2A_processor"] = "l2awinddirection"
         res.attrs["winddirection_L2A_processor_version"] = l2awinddirection.__version__
-        res.attrs["used_channel"] = res.coords["pol"].values
+        if 'pol' in res.coords:
+            res.attrs["used_channel"] = res.coords["pol"].values
+        else:
+            res.attrs['used_channel'] = res['selected_pol'].values
+            # res.attrs["used_channel"] =
         res.attrs["winddirection_L2A_processor_mode"] = args.mode
         if 'pol' in res.coords:
             res = res.drop('pol')