Update data/data_params.py

3778 · Mar 22, 2020 · 9ad9275 · 9ad9275
1 parent c5af483
commit 9ad9275
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/data/data_params.py b/data/data_params.py
@@ -150,7 +150,7 @@ def estimate_E0(value: 'query uf/city value',
     if granularity == 'Estado':
         E0 = (load_uf_covid_data()
               .query('uf == @value')
-              .assign(exposed=lambda df: df.new_cases
+              .assign(exposed=lambda df: df.cases
                                            .shift(-avg_incubation_time)
                                            .fillna(method='ffill')
                                            .fillna(0))  
@@ -188,4 +188,4 @@ def query_params(value: 'query uf/city value',
     E0 = estimate_E0(value, date, granularity) # temporary workaround
     I0 = query_I0(value, date, granularity)
     R0 = estimate_R0(value, date, granularity)
-    return N, E0, I0, R0
+    return N, E0, I0, R0