Fix dtype when reading data from json

- fix e.g. reading model '1' and '1.0' as float (leading to duplicate error)
IAMconsortium · Dec 11, 2019 · cf4d398 · cf4d398
1 parent 7d792c4
commit cf4d398
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pyam/iiasa.py b/pyam/iiasa.py
@@ -344,7 +344,9 @@ def query(self, **kwargs):
         r = requests.post(url, headers=headers, data=data)
         _check_response(r)
         # refactor returned json object to be castable to an IamDataFrame
-        df = pd.read_json(r.content, orient='records')
+        dtype = dict(model=str, scenario=str, variable=str, unit=str,
+                     region=str, year=int, value=float, version=int)
+        df = pd.read_json(r.content, orient='records', dtype=dtype)
         logger.debug('Response size is {0} bytes, '
                        '{1} records'.format(len(r.content), len(df)))
         columns = ['model', 'scenario', 'variable', 'unit',