fix(base_agents): fix Mesa model instantiation after upstream change

Met4FoF · May 14, 2024 · 3f25f43 · 3f25f43
1 parent bbb84ee
commit 3f25f43
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agentMET4FOF/agents/base_agents.py b/agentMET4FOF/agents/base_agents.py
@@ -10,7 +10,7 @@
 import matplotlib.pyplot as plt
 import mpld3
 import numpy as np
-from mesa import Agent as MesaAgent
+from mesa import Agent as MesaAgent, Model as MesaModel
 from osbrain import Agent as osBrainAgent
 from plotly import tools as tls
 from plotly.graph_objs import Scatter
@@ -45,7 +45,7 @@ def __init__(
         transport=None,
         attributes=None,
         backend=Backend.OSBRAIN,
-        mesa_model=None,
+        mesa_model=MesaModel,
     ):
         self.backend = self.validate_backend(backend)