fix: check if proton current coord exists

scipp · Dec 17, 2024 · a112f15 · a112f15
1 parent 4a373d1
commit a112f15
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/ess/amor/conversions.py b/src/ess/amor/conversions.py
@@ -161,10 +161,11 @@ def add_masks(
         wbins[0],
         wbins[-1],
     )
-    # Take inverse and use >= because we want to mask nan values
-    da.bins.masks['proton_current_too_low'] = ~(
-        da.bins.coords['proton_current'] >= da.coords['median_proton_current'] / 2
-    )
+    if 'proton_current' in da.bins.coords:
+        # Take inverse and use >= because we want to mask nan values
+        da.bins.masks['proton_current_too_low'] = ~(
+            da.bins.coords['proton_current'] >= da.coords['median_proton_current'] / 2
+        )
     return da