imitation.rewards.reward_function#

Type alias shared by reward-related code.

Classes

RewardFn(*args, **kwargs)

Abstract class for reward function.
class imitation.rewards.reward_function.RewardFn(*args, **kwargs)[source]#

Bases: Protocol

Abstract class for reward function.

Requires implementation of __call__() to compute the reward given a batch of states, actions, next states and dones.

__init__(*args, **kwargs)#