fix AttributeError when creating geodataframe from geoagents

projectmesa · Mar 23, 2024 · c307dbc · c307dbc
1 parent a256729
commit c307dbc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/mesa_geo/geospace.py b/mesa_geo/geospace.py
@@ -458,5 +458,9 @@ def get_agents_as_GeoDataFrame(self, agent_cls=GeoAgent) -> gpd.GeoDataFrame:
                 }
                 agents_list.append(agent_dict)
         agents_gdf = gpd.GeoDataFrame.from_records(agents_list, index="unique_id")
+        # workaround for geometry column not being set in `from_records`
+        # see https://github.com/geopandas/geopandas/issues/3152
+        # may be removed when the issue is resolved
+        agents_gdf.set_geometry("geometry", inplace=True)
         agents_gdf.crs = crs
         return agents_gdf
diff --git a/tests/test_GeoSpace.py b/tests/test_GeoSpace.py
@@ -143,6 +143,10 @@ def test_get_agents_as_GeoDataFrame(self):
             for agent in self.agents
         ]
         agents_gdf = gpd.GeoDataFrame.from_records(agents_list, index="unique_id")
+        # workaround for geometry column not being set in `from_records`
+        # see https://github.com/geopandas/geopandas/issues/3152
+        # may be removed when the issue is resolved
+        agents_gdf.set_geometry("geometry", inplace=True)
         agents_gdf.crs = self.geo_space.crs
 
         pd.testing.assert_frame_equal(