updating this example for new vector migration parameters

EMOD-Hub · Oct 30, 2024 · bf01fc1 · bf01fc1
1 parent a598fe2
commit bf01fc1
Show file tree

Hide file tree

Showing 7 changed files with 13 additions and 522 deletions.
diff --git a/emodpy_malaria/vector_migration/vector_migration.py b/emodpy_malaria/vector_migration/vector_migration.py
@@ -817,13 +817,7 @@ def _compute_migration_dict(node_list: list, gravity_params: list, exclude_nodes
         filename = f"vector_migration_{migration_type}.bin"
     v_migration.to_file(Path(filename))
     # turn on migration knobs in config
-    task.config.parameters.Enable_Vector_Migration = 1
-    if migration_type == VectorMigration.REGIONAL_MIGRATION:
-        task.config.parameters.Enable_Vector_Migration_Regional = 1
-        task.config.parameters.Vector_Migration_Filename_Regional = filename
-    elif migration_type == VectorMigration.LOCAL_MIGRATION:
-        task.config.parameters.Enable_Vector_Migration_Local = 1
-        task.config.parameters.Vector_Migration_Filename_local = filename
+    task.config.parameters.Vector_Species_Params[0].Vector_Migration_Filename = filename
     # add migration file to experiment
     task.common_assets.add_asset(filename)
     task.common_assets.add_asset(f"{filename}.json")