Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jun 20, 2024
1 parent 2247356 commit dae48c1
Showing 1 changed file with 9 additions and 13 deletions.
22 changes: 9 additions & 13 deletions scripts/shapes/units.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,14 +45,12 @@ def _build_layer(
resolution_config
"""
crs = [layer.crs for layer in source_layers.values()][0]
layer = pd.concat(
[
source_layers[source_layer][
source_layers[source_layer].country_code == _iso3(country)
]
for country, source_layer in resolution_config.items()
layer = pd.concat([
source_layers[source_layer][
source_layers[source_layer].country_code == _iso3(country)
]
)
for country, source_layer in resolution_config.items()
])
assert isinstance(layer, pd.DataFrame)
return gpd.GeoDataFrame(layer, crs=crs).reset_index(drop=True)

Expand Down Expand Up @@ -85,17 +83,15 @@ def _validate_resolution_config(
):
missing_countries = set(all_countries).difference(resolution_config)

assert (
not missing_countries
), f"Missing countries in {resolution} resolution configuration: {missing_countries}"
assert not missing_countries, f"Missing countries in {resolution} resolution configuration: {missing_countries}"


def _verify_all_countries_captured(
countries: list[str], units: gpd.GeoDataFrame, resolution: str
):
missing_countries = set(units.country_code).difference(
[_iso3(country) for country in countries]
)
missing_countries = set(units.country_code).difference([
_iso3(country) for country in countries
])
assert (
not missing_countries
), f"Countries are missing in {resolution} resolution shapes: {missing_countries}"
Expand Down

0 comments on commit dae48c1

Please sign in to comment.