diff --git a/digiplan/map/choropleths.py b/digiplan/map/choropleths.py index 6cc644d3..d497c9b1 100644 --- 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