Merge pull request #8 from IMMM-SFA/region-name

Added clause for extractor output
JGCRI · Feb 6, 2019 · 75ab68d · 75ab68d
2 parents 35ba7c2 + 6807273
commit 75ab68d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/demeter/demeter_io/reader.py b/demeter/demeter_io/reader.py
@@ -215,6 +215,12 @@ def read_gcam_file(log, f, gcam_landclasses, start_yr, end_yr, scenario, region_
     # assign user-defined scenario to data frame
     gdf['scenario'] = scenario
 
+    # check for extractor output which makes region ... gcam_region_name
+    try:
+        gdf['region'] = gdf['gcam_region_name']
+    except KeyError:
+        pass
+
     # create a list of GCAM years from header that are within the user specified year range
     user_years = _get_steps(gdf, start_yr, end_yr)