MetaWorld Wrappers

Descriptions

This module provides a Gymnasium wrapper for Meta-World environments that replaces the environment’s reward with the binary success signal. The reward will be either -1.0 (failure) or 0.0 (success).

Classes

class objectrl.utils.environment.metaworld_wrappers.SparsifyRewardWrapper(env)

Bases: Wrapper

Gymnasium wrapper for Meta-World environments that replaces reward with the binary success signal.

Parameters:

env – gym.Env The Meta-World environment to wrap.

__init__(env)

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

Parameters:

env – The environment to wrap

step(action)

Uses the step() of the env that can be overwritten to change the returned data.