iclbench.environments.babyai_text package

Submodules

iclbench.environments.babyai_text.clean_lang_wrapper module

class iclbench.environments.babyai_text.clean_lang_wrapper.BabyAITextCleanLangWrapper(env, vlm=False, **kwargs)

Bases: Wrapper

__init__(env, vlm=False, **kwargs)

Wraps an environment to allow a modular transformation of the step() and reset() methods.

Parameters:

env – The environment to wrap

property default_action

get_prompt(obs, infos)

get_stats()

get_text_action(action)

property interleaving_token

reset()

Resets the environment with kwargs.

step(action)

Steps through the environment with action.

Module contents

iclbench.environments.babyai_text.get_instruction_prompt(env, mission='BabyAI-MixedTrainLocal-v0')