wb data now removes NaN countries

gidden · Aug 21, 2017 · dae729f · dae729f
1 parent 1d5914a
commit dae729f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/salamanca/data.py b/salamanca/data.py
@@ -157,6 +157,9 @@ def query(self, indicator, tries=5, use_cache=True, overwrite=False, **kwargs):
             df['date'] = df['date'].astype(int)
         except:
             pass
+        # remove NaN countries
+        # TODO: why are there NaNs?
+        df = df.dropna(subset=['country'])
 
         # write to disc if we're caching
         if use_cache and (not exists or overwrite):