Basic mutli-agent environment implementation

mikelma · Nov 14, 2024 · 83e236d · 83e236d
1 parent b032a54
commit 83e236d
Show file tree

Hide file tree

Showing 7 changed files with 641 additions and 9 deletions.
diff --git a/craftium/__init__.py b/craftium/__init__.py
@@ -1,4 +1,5 @@
 from .craftium_env import CraftiumEnv
+from .multiagent_env import MarlCraftiumEnv
 from .wrappers import BinaryActionWrapper, DiscreteActionWrapper
 
 from gymnasium.envs.registration import register, WrapperSpec

diff --git a/craftium/craftium_env.py b/craftium/craftium_env.py
@@ -144,7 +144,7 @@ def reset(
         self.timesteps = 0
 
         if self.mt_chann.is_open():
-            self.mt_chann.send_termination()
+            self.mt_chann.send_kill()
             self.mt_chann.close_conn()
             self.mt.close_pipes()
             self.mt.wait_close()
@@ -225,7 +225,7 @@ def render(self):
 
     def close(self):
         if self.mt_chann.is_open():
-            self.mt_chann.send_termination()
+            self.mt_chann.send_kill()
             self.mt_chann.close()
             self.mt.close_pipes()
             self.mt.wait_close()