sinergym.utils.common.unwrap_wrapper

sinergym.utils.common.unwrap_wrapper(env: Env, wrapper_class: Type[Wrapper]) → Wrapper | None

Retrieve a wrapper object by recursively searching.

Parameters:

  • env – Environment to unwrap

  • wrapper_class – Wrapper to look for

Returns:

Environment unwrapped till wrapper_class if it has been wrapped with it