openai · DrArtemi · Jan 12, 2021 · Jan 13, 2021 · Jan 13, 2021
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,4 @@
 __pycache__/
 *.egg-info/
-*.pyc
+*.pyc
+.vscode/
diff --git a/bin/interactive.py b/bin/interactive.py
@@ -3,7 +3,7 @@
 sys.path.insert(1, os.path.join(sys.path[0], '..'))
 import argparse
 
-from multiagent.environment import MultiAgentEnv
+from multiagent.envs.environment import MultiAgentEnv
 from multiagent.policy import InteractivePolicy
 import multiagent.scenarios as scenarios
 

diff --git a/make_env.py b/make_env.py
diff --git a/multiagent/__init__.py b/multiagent/__init__.py
@@ -5,14 +5,5 @@
 
 register(
     id='MultiagentSimple-v0',
-    entry_point='multiagent.envs:SimpleEnv',
-    # FIXME(cathywu) currently has to be exactly max_path_length parameters in
-    # rllab run script
-    max_episode_steps=100,
-)
-
-register(
-    id='MultiagentSimpleSpeakerListener-v0',
-    entry_point='multiagent.envs:SimpleSpeakerListenerEnv',
-    max_episode_steps=100,
+    entry_point='multiagent.envs:MultiAgentEnv',
 )
diff --git a/multiagent/core.py b/multiagent/core.py