Fix pop and pop density choropleths

rl-institut · Oct 19, 2023 · 1d5b9d2 · 1d5b9d2
1 parent 1ae3f1c
commit 1d5b9d2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/digiplan/map/choropleths.py b/digiplan/map/choropleths.py
@@ -153,14 +153,14 @@ def get_values_per_feature(self) -> dict[int, float]:  # noqa: D102
 
 class PopulationChoropleth(Choropleth):  # noqa: D101
     def get_values_per_feature(self) -> dict[int, float]:  # noqa: D102
-        return models.Population.quantity_per_municipality_per_year().sum(axis=1).to_dict()
+        return models.Population.quantity_per_municipality_per_year()[2022].to_dict()
 
 
 class PopulationDensityChoropleth(Choropleth):  # noqa: D101
     def get_values_per_feature(self) -> dict[int, float]:  # noqa: D102
-        population = models.Population.quantity_per_municipality_per_year()
+        population = models.Population.quantity_per_municipality_per_year()[2022]
         population_square = calculations.calculate_square_for_value(population)
-        return population_square.sum(axis=1).to_dict()
+        return population_square.to_dict()
 
 
 class EmployeesChoropleth(Choropleth):  # noqa: D101