Commit b8f59603 authored by nilabha's avatar nilabha

Remove unused attributes and update config

parent a9c2418d
......@@ -51,7 +51,6 @@ class FlatlandSparse(MultiAgentEnv):
def __init__(self, env_config) -> None:
super().__init__()
self.video_folders = []
# TODO implement other generators
assert env_config['generator'] == 'sparse_rail_generator'
......@@ -140,7 +139,6 @@ class FlatlandSparse(MultiAgentEnv):
folder = self._env_config.get('video_dir',env_name)
monitor.Monitor._after_step =_after_step
self._env = monitor.Monitor(self._env, folder, resume=True)
self.video_folders.append(folder)
return self._env.reset()
def render(self,mode='human'):
......
......@@ -44,7 +44,6 @@ flatland-render-test:
project: neurips2020-flatland-baselines
entity: nilabha2007
tags: ["small_v0", "tree_obs"] # TODO should be set programmatically
reinit: True
# monitor_gym: True # Wandb video doesn't seem to work
model:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment