datasets.RTFM package

Submodules

datasets.RTFM.get_env module

Implements environment getter for RTFM.

datasets.RTFM.get_env.create_env(env='rtfm:groups_simple_stationary-v0', height=6, width=6, partially_observable=False, max_placement=1, featurizer=None, shuffle_wiki=False, time_penalty=-0.02)

Create RTFM environment.

Parameters:

  • env (str, optional) – RTFM environment name.. Defaults to “rtfm:groups_simple_stationary-v0”.

  • height (int, optional) – Height of environment. Defaults to 6.

  • width (int, optional) – Width of environment. Defaults to 6.

  • partially_observable (bool, optional) – Whether to only give partial observations or priviledged observations. Defaults to False.

  • max_placement (int, optional) – Max placement. Defaults to 1.

  • featurizer (_type_, optional) – Function for featurizing inputs. Defaults to None.

  • shuffle_wiki (bool, optional) – Whether to shuffle wiki. Defaults to False.

  • time_penalty (float, optional) – Time penalty. Defaults to -0.02.

Returns:

gym environment

Return type:

env

Module contents