Refactor the list interface MAE wrapper.

CDM1619 · Nov 11, 2021 · f218ead · f218ead
1 parent cfcbbfb
commit f218ead
Show file tree

Hide file tree

Showing 6 changed files with 115 additions and 357 deletions.
diff --git a/examples/marl/openai/train.py b/examples/marl/openai/train.py
@@ -28,7 +28,7 @@
 
 from gridworld.log import logger
 from gridworld.multiagent_env import MultiAgentEnv
-from gridworld.multiagent_env_openai_style import MultiagentEnvOpenAIStyle
+from gridworld.multiagent_list_interface_env import MultiAgentListInterfaceEnv
 from gridworld.scenarios.buildings import make_env_config
 
 logger.setLevel(logging.ERROR)
@@ -180,7 +180,7 @@ def make_env(arglist):
         num_buildings=3
     )
 
-    env = MultiagentEnvOpenAIStyle(
+    env = MultiAgentListInterfaceEnv(
         CoordinatedMultiBuildingControlEnv,
         env_config
     )

diff --git a/gridworld/__init__.py b/gridworld/__init__.py
@@ -1,4 +1,4 @@
 __version__ = "0.0.1"
 from .base import ComponentEnv, MultiComponentEnv
 from .multiagent_env import MultiAgentEnv
-from .multiagent_env_openai_style import MultiagentEnvOpenAIStyle
+from .multiagent_list_interface_env import MultiAgentListInterfaceEnv
diff --git a/gridworld/multiagent_env_citylearn_style.py b/gridworld/multiagent_env_citylearn_style.py
diff --git a/gridworld/multiagent_env_openai_style.py b/gridworld/multiagent_env_openai_style.py