Added offsets to CorrectLevel1Registrations.

djgroen · Nov 13, 2023 · 53aba7b · 53aba7b
1 parent d4d5daf
commit 53aba7b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flee/datamanager/handle_refugee_data.py b/flee/datamanager/handle_refugee_data.py
@@ -71,8 +71,8 @@ def correctLevel1Registrations(self, name: str, date: str) -> float:
             if int(ref_table[i][0]) == int(days):
                 # then scale all previous entries by ref_table[i][1]/ref_table[i-1][1]
                 if i > 0:
-                    first_level_2_value = ref_table[i, 1]
-                    last_level_1_value = ref_table[i - 1, 1]
+                    first_level_2_value = ref_table[i, 1] + self.offsets.get(name, 0)
+                    last_level_1_value = ref_table[i - 1, 1] + self.offsets.get(name, 0)
                     # print(days, i, ref_table[0:i,1])
                     ref_table[0:i, 1] *= first_level_2_value / last_level_1_value
                     # print(first_level_2_value, last_level_1_value, ref_table[0:i,1])