From 9aa153c687299709eb625639082e04f79822ee2f Mon Sep 17 00:00:00 2001 From: Tim Plummer Date: Mon, 18 May 2026 08:59:20 -0600 Subject: [PATCH] Fix issue with cdf attributes in Lo l1a datasets --- imap_processing/lo/l1a/lo_l1a.py | 8 ++++++-- imap_processing/tests/lo/test_lo_l1a.py | 4 ++++ 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/imap_processing/lo/l1a/lo_l1a.py b/imap_processing/lo/l1a/lo_l1a.py index 21c0acfa6a..6c24c7998d 100644 --- a/imap_processing/lo/l1a/lo_l1a.py +++ b/imap_processing/lo/l1a/lo_l1a.py @@ -243,7 +243,9 @@ def add_dataset_attrs( data=azimuth_60.values.astype(str), name="azimuth_60_label", dims=["azimuth_60_label"], - attrs=attr_mgr.get_variable_attributes("azimuth_60_label"), + attrs=attr_mgr.get_variable_attributes( + "azimuth_60_label", check_schema=False + ), ) azimuth_6 = xr.DataArray( data=np.arange(0, 60, dtype=np.uint8), @@ -255,7 +257,9 @@ def add_dataset_attrs( data=azimuth_6.values.astype(str), name="azimuth_6_label", dims=["azimuth_6_label"], - attrs=attr_mgr.get_variable_attributes("azimuth_6_label"), + attrs=attr_mgr.get_variable_attributes( + "azimuth_6_label", check_schema=False + ), ) esa_step = xr.DataArray( diff --git a/imap_processing/tests/lo/test_lo_l1a.py b/imap_processing/tests/lo/test_lo_l1a.py index d20c93064d..1f98a3655a 100644 --- a/imap_processing/tests/lo/test_lo_l1a.py +++ b/imap_processing/tests/lo/test_lo_l1a.py @@ -2,6 +2,7 @@ import pandas as pd from imap_processing import imap_module_directory +from imap_processing.cdf.utils import write_cdf from imap_processing.lo.l1a.lo_l1a import lo_l1a @@ -44,6 +45,9 @@ def test_lo_l1a(): for dataset, logical_source in zip( output_dataset, expected_logical_source, strict=False ): + # Try writing out the dataset to cdf in an attempt to catch any issues + # with attributes that cdflib doesn't like + _ = write_cdf(dataset) assert logical_source == dataset.attrs["Logical_source"] for var in dataset: if var in no_depend_0_vars or var.endswith("label"):