def make()

in jat/eval/rl/core.py [0:0]

• 11 lines of code
• 5 McCabe index (conditional complexity)

def make(task_name: str, **kwargs) -> Env:
    """
    Make an environment from the task name.

    Args:
        task_name (`str`):
            The name of the task to make. Check `get_task_names()` for the list of available tasks.

    Raises:
        ValueError:
            If the task name is not in the list of available tasks.

    Returns:
        Env: The environment.
    """
    if task_name.startswith("atari"):
        return make_atari(task_name, **kwargs)

    elif task_name.startswith("babyai"):
        return make_babyai(task_name, **kwargs)

    elif task_name.startswith("metaworld"):
        return make_metaworld(task_name, **kwargs)

    elif task_name.startswith("mujoco"):
        return make_mujoco(task_name, **kwargs)
    else:
        raise ValueError(f"Unknown task name: {task_name}. Available task names: {get_task_names()}")