Trait RewardTransform 

pub trait RewardTransform: Send + Sync {
    // Required methods
    fn transform(
        &self,
        rewards: &[f64],
        context: &RewardContext<'_>,
    ) -> Result<Vec<f64>, RloxError>;
    fn name(&self) -> &str;
}

Trait for composable reward transformations.

Each transform takes raw rewards and a context and produces shaped rewards. Transforms can be chained: ClipReward -> PBRSTransform -> ScaleReward.

Required Methods

fn transform( &self, rewards: &[f64], context: &RewardContext<'_>, ) -> Result<Vec<f64>, RloxError>

Transform a batch of rewards.

fn name(&self) -> &str

Human-readable name.

Implementors

impl RewardTransform for GoalDistanceTransform

impl RewardTransform for PBRSTransform