a lot of fixes

2023-06-26 11:31:47 +02:00
parent 966db6e5ad
commit f60dca6367
23 changed files with 338 additions and 45 deletions
--- a/econ/agents/base_agent.py
+++ b/econ/agents/base_agent.py
@@ -23,4 +23,16 @@ class BaseAgent(ABC):
        Resets agent to new episode.
        """
        assert "No reset method has been provided"
-    pass
+    pass
+
+    def unregister(self):
+        """
+        Disables agent by removing it from the simulation
+        """
+        self.simulation.unregister_agent(id)
+    def register(self):
+        """
+        Enables agent by adding it to the simulation:
+        """
+        self.simulation.register_agent(self.id,self.tick,self.reset)
+