Fix small bug in _parse_geo_response and add some doc strings

mggg · Jun 25, 2024 · 9413ead · 9413ead
1 parent 5ac3752
commit 9413ead
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/gerrydb/repos/geography.py b/gerrydb/repos/geography.py
@@ -64,6 +64,7 @@ def _parse_geo_response(response: httpx.Response) -> list[Geography]:
     response_geos = []
     for response_geo in msgpack.loads(response.content):
         response_geo["geography"] = shapely.wkb.loads(response_geo["geography"])
+        response_geo["internal_point"] = shapely.wkb.loads(response_geo["internal_point"])
         response_geos.append(Geography(**response_geo))
     return response_geos
 

diff --git a/gerrydb/repos/graph.py b/gerrydb/repos/graph.py
@@ -31,7 +31,8 @@ def create(
         description: str,
         proj: Optional[str] = None,
     ) -> Graph:
-        """Imports a dual graph from a NetworkX graph.
+        """
+        Imports a dual graph from a NetworkX graph.
 
         Args:
             path: A short identifier for the graph (e.g. `iowa_counties_rook`).
@@ -49,7 +50,8 @@ def create(
                 or if the parameters fail validation.
 
         Returns:
-            The new districting plan.
+            The new districting plan in the form of a gerrydb `Graph` schema
+            object.
         """
         response = self.ctx.client.post(
             f"{self.base_url}/{namespace}",