FIX set value for AI4GCC unittests

pjwozny · Mar 2, 2023 · a8e2878 · a8e2878
1 parent 6ef02c1
commit a8e2878
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/rice.py b/rice.py
@@ -105,6 +105,10 @@ def __init__(
         params, self.num_regions = set_rice_params(
             Path(_PUBLIC_REPO_DIR) / "region_yamls"
         )
+
+        # must be set for AI4GCC unittests
+        self.num_agents = self.num_regions
+
         # TODO : add to yaml
         self.balance_interest_rate = 0.1