bug fixes and reorganization of ensemble analysis

gcrsef-gang · Feb 25, 2023 · d3cd102 · d3cd102
1 parent 174be91
commit d3cd102
Show file tree

Hide file tree

Showing 7 changed files with 411 additions and 155 deletions.
diff --git a/rba/__main__.py b/rba/__main__.py
@@ -56,7 +56,11 @@
     ensemble_parser.add_argument("--graph_file", type=str, default=os.path.join(package_dir, "data/2010/new_hampshire_geodata_merged.json"))
     ensemble_parser.add_argument("--community_file", type=str, default=os.path.join(package_dir, "data/2010/new_hampshire_communities.json"))
     ensemble_parser.add_argument("--vra_config_file", type=str, default=os.path.join(package_dir, "data/2010/vra_nh.json"))
+    ensemble_parser.add_argument("--num_steps", type=int, default=100)
+    ensemble_parser.add_argument("--num_districts", type=int, default=2)
+    ensemble_parser.add_argument("--initial_plan_file", type=str, default=None)
     ensemble_parser.add_argument("--district_file", type=str, default=os.path.join(package_dir, "data/2010/new_hampshire_districts.json"))
+    ensemble_parser.add_argument("-o", "--output_dir", type=str)
     ensemble_parser.set_defaults(func=rba.ensemble.ensemble_analysis)
 
     optimize_parser = subparsers.add_parser("optimize")

diff --git a/rba/data/2010/vra_ma.json b/rba/data/2010/vra_ma.json