fix(bbox): remove the bbox in predictions

hotosm · Nov 26, 2024 · 73ac1cd · 73ac1cd
1 parent 7e151b3
commit 73ac1cd
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/predictor/app.py b/predictor/app.py
@@ -97,6 +97,15 @@ def predict(
         prediction_geojson_data = json.load(f)
     if remove_metadata:
         shutil.rmtree(base_path)
+
+    if (
+        "features" in prediction_geojson_data
+        and len(prediction_geojson_data["features"]) > 0
+    ):
+        prediction_geojson_data["features"].pop(
+            -1
+        )  # Remove the last feature explicitly for the removal of whole bbox
+
     for feature in prediction_geojson_data["features"]:
         feature["properties"]["building"] = "yes"
         feature["properties"]["source"] = "fAIr"