thing

projectmesa · Sep 20, 2024 · b9d3559 · b9d3559
1 parent b734ada
commit b9d3559
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 42 deletions.
diff --git a/mesa/__init__.py b/mesa/__init__.py
@@ -8,18 +8,13 @@
 import mesa.space as space
 import mesa.time as time
 from mesa.agent import Agent
-from mesa.batchrunner import batch_run
-from mesa.datacollection import DataCollector
 from mesa.model import Model
 
 __all__ = [
     "Model",
     "Agent",
     "time",
     "space",
-    "DataCollector",
-    "batch_run",
-    "experimental",
 ]
 
 __title__ = "mesa"

diff --git a/tests/test_batch_run.py b/tests/test_batch_run.py
@@ -1,8 +1,7 @@
 """Test Batchrunner."""
 
-import mesa
 from mesa.agent import Agent
-from mesa.batchrunner import _make_model_kwargs
+from mesa.batchrunner import _make_model_kwargs, batch_run
 from mesa.datacollection import DataCollector
 from mesa.model import Model
 from mesa.time import BaseScheduler
@@ -103,7 +102,7 @@ def step(self):  # noqa: D102
 
 
 def test_batch_run():  # noqa: D103
-    result = mesa.batch_run(MockModel, {}, number_processes=2)
+    result = batch_run(MockModel, {}, number_processes=2)
     assert result == [
         {
             "RunId": 0,
@@ -136,7 +135,7 @@ def test_batch_run():  # noqa: D103
 
 
 def test_batch_run_with_params():  # noqa: D103
-    mesa.batch_run(
+    batch_run(
         MockModel,
         {
             "variable_model_params": range(3),
@@ -147,9 +146,7 @@ def test_batch_run_with_params():  # noqa: D103
 
 
 def test_batch_run_no_agent_reporters():  # noqa: D103
-    result = mesa.batch_run(
-        MockModel, {"enable_agent_reporters": False}, number_processes=2
-    )
+    result = batch_run(MockModel, {"enable_agent_reporters": False}, number_processes=2)
     print(result)
     assert result == [
         {
@@ -163,11 +160,11 @@ def test_batch_run_no_agent_reporters():  # noqa: D103
 
 
 def test_batch_run_single_core():  # noqa: D103
-    mesa.batch_run(MockModel, {}, number_processes=1, iterations=6)
+    batch_run(MockModel, {}, number_processes=1, iterations=6)
 
 
 def test_batch_run_unhashable_param():  # noqa: D103
-    result = mesa.batch_run(
+    result = batch_run(
         MockModel,
         {
             "n_agents": 2,

diff --git a/tests/test_import_namespace.py b/tests/test_import_namespace.py