Skip to content

Commit 89b6af7

Browse files
fix: babyai read kwargs
1 parent 67a8d26 commit 89b6af7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

balrog/environments/babyai_text/babyai_env.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def make_babyai_env(env_name, task, config, render_mode: Optional[str] = None):
3838
if task.startswith("BabyAI-MixedTrainLocal-v0/"):
3939
base_task, goal = task.split("/")
4040
while 1:
41-
env = gym.make(base_task, render_mode=render_mode)
41+
env = gym.make(base_task, **config.envs.babyai_kwargs)
4242
if env.unwrapped.action_kinds[0].replace(" ", "_") == goal:
4343
break
4444

0 commit comments

Comments
 (0)