From f872798a39c6681beb10a3ffc6b28f29b370033e Mon Sep 17 00:00:00 2001 From: syedhamidali Date: Sat, 31 Aug 2024 18:11:29 -0400 Subject: [PATCH] FIX: Cfradial1 writer --- xradar/io/export/cfradial1.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/xradar/io/export/cfradial1.py b/xradar/io/export/cfradial1.py index af93c660..8d1888f4 100644 --- a/xradar/io/export/cfradial1.py +++ b/xradar/io/export/cfradial1.py @@ -76,11 +76,12 @@ def _main_info_mapper(dtree): xarray.Dataset Dataset containing the mapped radar information. """ - dataset = ( - dtree.root.to_dataset() - .drop_vars("sweep_group_name", errors="ignore") - .rename({"sweep_fixed_angle": "fixed_angle"}) - ) + dataset = dtree.root.to_dataset().drop_vars("sweep_group_name", errors="ignore") + + # Only rename 'sweep_fixed_angle' if it exists in the dataset + if "sweep_fixed_angle" in dataset.variables: + dataset = dataset.rename({"sweep_fixed_angle": "fixed_angle"}) + return dataset @@ -102,7 +103,7 @@ def _variable_mapper(dtree, dim0=None): """ sweep_info = _sweep_info_mapper(dtree) - vol_info = _main_info_mapper(dtree).drop_vars("fixed_angle") + vol_info = _main_info_mapper(dtree).drop_vars("fixed_angle", errors='ignore') sweep_datasets = [] for grp in dtree.groups: if "sweep" in grp: