avoid crashing when census site is down

Deltares · Nov 1, 2024 · f3cf4ab · f3cf4ab
1 parent ccbe70f
commit f3cf4ab
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/hydromt_fiat/fiat.py b/hydromt_fiat/fiat.py
@@ -909,7 +909,11 @@ def setup_equity_data(
         equity.download_census_data(year_data)
         equity.rename_census_data()
         equity.match_geo_ID()
-        equity.download_shp_geom(year_data, county_numbers)
+        try:
+            equity.download_shp_geom(year_data, county_numbers)
+        except:
+            self.logger.warning("The census track shapefile could not be downloaded, potentially because the site is down. Aggregation areas and equity information will not be available in the FIAT model!")
+            return
         equity.merge_equity_data_shp()
         equity.clean()
         gdf = rename_geoid_short(equity.equity_data_shp)