Fix to MPI tests.

djgroen · Nov 3, 2023 · ed9799d · ed9799d
1 parent 33abc16
commit ed9799d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/flee/pflee.py b/flee/pflee.py
@@ -149,6 +149,7 @@ def __init__(
         e,
         cur_id: int,
         name: str,
+        region: str = "unknown",
         x: float = 0.0,
         y: float = 0.0,
         location_type: Optional[str] = None,
@@ -166,6 +167,7 @@ def __init__(
         # If it is referred to in Flee in any way, the code should crash.
         super().__init__(
             name=name,
+            region=region,
             x=x,
             y=y,
             location_type=location_type,
@@ -665,6 +667,7 @@ def evolve(self) -> None:
     def addLocation(
         self,
         name: str,
+        region: str = "unknown",
         x: float = 0.0,
         y: float = 0.0,
         location_type: Optional[str] = None,
@@ -707,6 +710,7 @@ def addLocation(
             self,
             cur_id=self.cur_loc_id,
             name=name,
+            region=region,
             x=x,
             y=y,
             location_type=location_type,