Made modifications based on recent updates

NAU-CCL · Jan 25, 2024 · 37c0f93 · 37c0f93
1 parent 5fe4c45
commit 37c0f93
Show file tree

Hide file tree

Showing 4 changed files with 65 additions and 94 deletions.
diff --git a/doc/devlog/example_local_birds.ipynb b/doc/devlog/example_local_birds.ipynb
diff --git a/doc/devlog/example_long_birds.ipynb b/doc/devlog/example_long_birds.ipynb
diff --git a/doc/devlog/example_mosquito.ipynb b/doc/devlog/example_mosquito.ipynb
diff --git a/epymorph/initializer.py b/epymorph/initializer.py
@@ -270,11 +270,11 @@ def bottom_locations(ctx: InitContext, attribute: str, num_locations: int, seed_
     return indexed_locations(ctx, selection, seed_size)
 
 def bird_movement_initializer(ctx: InitContext)->NDArray[SimDType]:
-    result = ctx.geo['bird_population'].reshape((ctx.nodes, ctx.compartments)) # type: ignore
+    result = ctx.geo['bird_population'].reshape((ctx.dim.nodes, ctx.dim.compartments)) # type: ignore
     return result.astype(SimDType)
 
 def mosquito_movement_initializer(ctx: InitContext)->NDArray[SimDType]:
-    result = ctx.geo['bird_population'].reshape((ctx.nodes, ctx.compartments)) # type: ignore
+    result = ctx.geo['bird_population'].reshape((ctx.dim.nodes, ctx.dim.compartments)) # type: ignore
     return result.astype(SimDType)
 
 DEFAULT_INITIALIZER = single_location