check the input data for "unit"

iiasa · Apr 3, 2020 · e9d9430 · e9d9430
1 parent 9ca3acd
commit e9d9430
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/message_ix/macro.py b/message_ix/macro.py
@@ -124,7 +124,7 @@ def validate(kind, values, df):
         cols = MACRO_DATA_FOR_DERIVATION[name]
     else:
         cols = MACRO_ITEMS[name]['idx_sets']
-    col_diff = set(cols) - set(df.columns)
+    col_diff = set(cols + ['unit']) - set(df.columns)
     if col_diff:
         msg = 'Missing expected columns for {}: {}'
         raise ValueError(msg.format(name, col_diff))