Tolerate missing labels in MappingAdapter.adapt

iiasa · Dec 4, 2024 · 8162acd · 8162acd
1 parent 63ddb71
commit 8162acd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/message_ix_models/util/common.py b/message_ix_models/util/common.py
@@ -118,8 +118,9 @@ def adapt(self, qty: Quantity) -> Quantity:
                 continue
             result = concat(
                 *[
-                    qty.sel({dim: label[0]}, drop=True).expand_dims({dim: [label[1]]})
-                    for label in labels
+                    qty.sel({dim: a}, drop=True).expand_dims({dim: [b]})
+                    for (a, b) in labels
+                    if a in qty.coords[dim]  # Skip `label` if not in `dim` of `qty`
                 ]
             )