minor changes

gcrsef-gang · Feb 22, 2023 · 192fd07 · 192fd07
1 parent db6ce24
commit 192fd07
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/rba/community_generation.py b/rba/community_generation.py
@@ -24,13 +24,6 @@
 }
 
 
-@dataclass
-class Community:
-    precincts: frozenset
-    birth_time: float
-    death_time: float = -1
-
-
 def compute_precinct_similarities(graph, verbose=False):
     """
     Generates similarities between precincts. Edits graph in-place.

diff --git a/rba/visualization.py b/rba/visualization.py
@@ -103,7 +103,7 @@ def modify_coords(coords, bounds):
     return new_coords
 
 
-def visualize_partition_geopandas(partition):
+def visualize_partition_geopandas(partition, *args, **kwargs):
     """Visualizes a gerrychain.Partition object using geopandas.
     """
     data = {"assignment": [], "geometry": []}
@@ -113,7 +113,7 @@ def visualize_partition_geopandas(partition):
 
     gdf = geopandas.GeoDataFrame(data)
     del data
-    gdf.plot(column="assignment")
+    gdf.plot(column="assignment", *args, **kwargs)
 
 
 def visualize_map(graph, output_fpath, node_coords, edge_coords, node_colors=None, edge_colors=None,