Update fr_covidata.py

CoronaWhy · Apr 10, 2020 · 5673326 · 5673326
1 parent 1f840b3
commit 5673326
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/task_geo/data_sources/covid/fr_covidata/fr_covidata.py b/task_geo/data_sources/covid/fr_covidata/fr_covidata.py
@@ -67,11 +67,10 @@ def fr_covidata_formatter(dataset):
         frcovidata(pandas.DataFrame)
     """
 
-    no_granularites = ['region', 'monde', 'pays', 'collectivite-outremer']
-    no_maille_codes = ['DEP-971', 'DEP-972', 'DEP-973', 'DEP-974', 'DEP-976']
+    no_gr = ['region', 'monde', 'pays', 'collectivite-outremer']
+    no_mc = ['DEP-971', 'DEP-972', 'DEP-973', 'DEP-974', 'DEP-976']
     dataset = dataset[
-        (~dataset.granularite.isin(no_granularites)) &
-        (~dataset.maille_code.isin(no_maille_codes))
+        (~dataset.granularite.isin(no_gr)) & (~dataset.maille_code.isin(no_mc))
     ]
     dataset = dataset.drop(['depistes', 'granularite'], axis=1)
     dataset = dataset.drop_duplicates(