Fix missing check that metrics are present

Urban-Analytics-Technology-Platform · Jul 1, 2024 · ab19647 · ab19647
1 parent 65fc4c0
commit ab19647
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/popgetter/assets/usa/__init__.py b/python/popgetter/assets/usa/__init__.py
@@ -430,7 +430,7 @@ def derived_metrics(
 
             # TODO: refactor asset
             variable_dictionary = generate_variable_dictionary(year, summary_level)
-            if census_tables.shape[0] == 0:
+            if census_tables.shape[0] == 0 or census_tables.shape[1] == 0:
                 context.log.warning(f"No metrics found in parition: {partition_key}")
                 return MetricsOutput(metadata=[], metrics=pd.DataFrame())