FIX: Update SWAPI quality flags to work with numpy 2+

IMAP-Science-Operations-Center · Sep 16, 2024 · d59b3de · d59b3de
1 parent 094afd8
commit d59b3de
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 42 deletions.
diff --git a/imap_processing/swapi/l1/swapi_l1.py b/imap_processing/swapi/l1/swapi_l1.py
@@ -474,9 +474,9 @@ def process_swapi_science(
     quality_flags_data = np.zeros((total_full_sweeps, 72), np.uint16)
 
     # Add science data quality flags
-    quality_flags_data[pcem_compression_flags == 1] |= SWAPIFlags.SWP_PCEM_COMP
-    quality_flags_data[scem_compression_flags == 1] |= SWAPIFlags.SWP_SCEM_COMP
-    quality_flags_data[coin_compression_flags == 1] |= SWAPIFlags.SWP_COIN_COMP
+    quality_flags_data[pcem_compression_flags == 1] |= SWAPIFlags.SWP_PCEM_COMP.value
+    quality_flags_data[scem_compression_flags == 1] |= SWAPIFlags.SWP_SCEM_COMP.value
+    quality_flags_data[coin_compression_flags == 1] |= SWAPIFlags.SWP_COIN_COMP.value
 
     # Add housekeeping-derived quality flags
     # --------------------------------------
@@ -519,7 +519,7 @@ def process_swapi_science(
         # Use getattr to dynamically access the flag in SWAPIFlags class
         flag_to_set = getattr(SWAPIFlags, flag_name)
         # set the quality flag for each data
-        quality_flags_data[current_flag == 1] |= flag_to_set
+        quality_flags_data[current_flag == 1] |= flag_to_set.value
 
     swp_flags = xr.DataArray(
         quality_flags_data,

diff --git a/poetry.lock b/poetry.lock