Struct TestEnv

pub struct TestEnv { /* private fields */ }

Environment for testing.

Trait Implementations

impl Agent<TestEnv, SimpleReplayBuffer<TestObsBatch, TestActBatch>> for TestAgent

fn train(&mut self)

Switches the agent to training mode.

fn is_train(&self) -> bool

Returns whether the agent is currently in training mode.

fn eval(&mut self)

Switches the agent to evaluation mode.

fn opt_with_record( &mut self, _buffer: &mut SimpleReplayBuffer<TestObsBatch, TestActBatch>, ) -> Record

Performs an optimization step and returns training metrics.

fn save_params(&self, _path: &Path) -> Result<Vec<PathBuf>>

Saves the agent’s parameters to the specified directory.

fn load_params(&mut self, _path: &Path) -> Result<()>

Loads the agent’s parameters from the specified directory.

fn as_any_ref(&self) -> &dyn Any

Returns a reference to the agent as a type-erased Any value.

fn as_any_mut(&mut self) -> &mut dyn Any

Returns a mutable reference to the agent as a type-erased Any value.

fn opt(&mut self, buffer: &mut R)

Performs a single optimization step using experiences from the replay buffer.

impl Env for TestEnv

type Config = usize

Configuration parameters for the environment.

type Obs = TestObs

The type of observations returned by the environment.

type Act = TestAct

The type of actions accepted by the environment.

type Info = TestInfo

Additional information returned with each step.

fn reset(&mut self, _is_done: Option<&Vec<i8>>) -> Result<Self::Obs>

Resets the environment to its initial state.

fn reset_with_index(&mut self, _ix: usize) -> Result<Self::Obs>

Resets the environment with a specific index.

fn step_with_reset(&mut self, a: &Self::Act) -> (Step<Self>, Record)

where Self: Sized,

Performs a step and automatically resets the environment if the episode ends.

fn step(&mut self, a: &Self::Act) -> (Step<Self>, Record)

where Self: Sized,

Performs a single step in the environment.

fn build(config: &Self::Config, _seed: i64) -> Result<Self>

where Self: Sized,

Builds a new instance of the environment with the given configuration and random seed.

impl Policy<TestEnv> for TestAgent

fn sample(&mut self, _obs: &TestObs) -> TestAct

Samples an action given an observation from the environment.