Struct VecEnv 

pub struct VecEnv { /* private fields */ }

A vectorized environment that steps multiple sub-environments in parallel.

Implementations

impl VecEnv

pub fn new(envs: Vec<Box<dyn RLEnv>>) -> Result<Self, RloxError>

Create a new vectorized environment from a list of sub-environments.

Errors

Returns RloxError::EnvError if envs is empty.

pub fn num_envs(&self) -> usize

pub fn action_space(&self) -> &ActionSpace

pub fn step_all( &mut self, actions: &[Action], ) -> Result<BatchTransition, RloxError>

Step all environments in parallel using Rayon.

If an environment is done after stepping, it is automatically reset and the returned observation is from the fresh episode.

pub fn step_all_flat( &mut self, actions: &[Action], ) -> Result<BatchTransition, RloxError>

Step all environments in parallel, returning observations as a flat contiguous buffer.

Unlike step_all, this avoids per-env Vec allocations by collecting observations directly into obs_flat: Vec<f32> of shape [n_envs * obs_dim]. The obs field is left empty.

pub fn reset_all( &mut self, seed: Option<u64>, ) -> Result<Vec<Observation>, RloxError>

Reset all environments, optionally seeding them deterministically.

When a master seed is provided, each env i gets derive_seed(master, i).

Trait Implementations

impl BatchSteppable for VecEnv

fn step_batch( &mut self, actions: &[Action], ) -> Result<BatchTransition, RloxError>

Step all environments with the given actions (one per env).

fn reset_batch( &mut self, seed: Option<u64>, ) -> Result<Vec<Observation>, RloxError>

Reset all environments, optionally seeding them deterministically.

fn num_envs(&self) -> usize

Number of sub-environments in this batch.

fn action_space(&self) -> &ActionSpace

The shared action space (all sub-environments must have the same space).

fn obs_space(&self) -> &ObsSpace

The shared observation space.

impl Debug for VecEnv

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.