From e4121a403a777867d540ec8d07b8c4f07980d02d Mon Sep 17 00:00:00 2001 From: Ewout ter Hoeven Date: Fri, 20 Sep 2024 14:41:24 +0200 Subject: [PATCH] docs: Update batch_run import in intro tutorial --- docs/overview.md | 2 +- docs/tutorials/intro_tutorial.ipynb | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/docs/overview.md b/docs/overview.md index be88f0315b6..93aac99c5eb 100644 --- a/docs/overview.md +++ b/docs/overview.md @@ -103,7 +103,7 @@ To batch-run the model while varying, for example, the n_agents parameter, you'd import mesa parameters = {"n_agents": range(1, 20)} -mesa.batch_run( +mesa.batchrunner.batch_run( MyModel, parameters, max_steps=10, diff --git a/docs/tutorials/intro_tutorial.ipynb b/docs/tutorials/intro_tutorial.ipynb index c2c83c894b5..ce40b3d3333 100644 --- a/docs/tutorials/intro_tutorial.ipynb +++ b/docs/tutorials/intro_tutorial.ipynb @@ -159,6 +159,7 @@ "outputs": [], "source": [ "import mesa\n", + "from mesa.batchrunner import batch_run\n", "from mesa.datacollection import DataCollector\n", "\n", "# Data visualization tools.\n", @@ -1166,7 +1167,7 @@ "source": [ "params = {\"width\": 10, \"height\": 10, \"N\": range(5, 100, 5)}\n", "\n", - "results = mesa.batch_run(\n", + "results = batch_run(\n", " MoneyModel,\n", " parameters=params,\n", " iterations=7,\n", @@ -1315,7 +1316,7 @@ "source": [ "params = {\"width\": 10, \"height\": 10, \"N\": [5, 10, 20, 40, 80]}\n", "\n", - "results_5s = mesa.batch_run(\n", + "results_5s = batch_run(\n", " MoneyModel,\n", " parameters=params,\n", " iterations=100,\n",