Trait StepProcessor

pub trait StepProcessor<E: Env> {
    type Config: Clone;
    type Output;

    // Required methods
    fn build(config: &Self::Config) -> Self;
    fn reset(&mut self, init_obs: E::Obs);
    fn process(&mut self, step: Step<E>) -> Self::Output;
}

Process Step and output an item Self::Output.

This trait is used in Trainer. Step object is transformed to Self::Output, which will be pushed into a replay buffer implementing ExperienceBufferBase. The type Self::Output should be the same with ExperienceBufferBase::Item.

Required Associated Types

type Config: Clone

Configuration.

type Output

The type of transitions produced by this trait.

Required Methods

fn build(config: &Self::Config) -> Self

Build a producer.

fn reset(&mut self, init_obs: E::Obs)

Resets the object.

fn process(&mut self, step: Step<E>) -> Self::Output

Processes a Step object.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl<E, O, A> StepProcessor<E> for SimpleStepProcessor<E, O, A>

where E: Env, O: BatchBase + From<E::Obs>, A: BatchBase + From<E::Act>,

type Config = SimpleStepProcessorConfig

type Output = GenericTransitionBatch<O, A>