fix renaming of geom column

NYCPlanning · Oct 2, 2024 · fffae94 · fffae94
1 parent 6868d9f
commit fffae94
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dcpy/lifecycle/ingest/transform.py b/dcpy/lifecycle/ingest/transform.py
@@ -97,6 +97,8 @@ def rename_columns(
         self, df: pd.DataFrame, map: dict[str, str], drop_others=False
     ) -> pd.DataFrame:
         renamed = df.copy()
+        if isinstance(renamed, gpd.GeoDataFrame) and renamed.geometry.name in map:
+            renamed.rename_geometry(map.pop(renamed.geometry.name), inplace=True)
         renamed = renamed.rename(columns=map, errors="raise")
         if drop_others:
             renamed = renamed[list(map.values())]