Struct AsyncCollector 

pub struct AsyncCollector { /* private fields */ }

Asynchronous rollout collector that runs env stepping and GAE computation in a background thread, sending completed batches through a channel.

The collector requires two external function pointers:

• action_fn: given flat observations [n_envs * obs_dim], returns (actions_flat, log_probs) where actions are [n_envs * act_dim] and log_probs are [n_envs].
• value_fn: given flat observations [n_envs * obs_dim], returns value estimates [n_envs].

This design lets the Python side own the neural network while Rust handles the tight env-stepping loop.

Implementations

impl AsyncCollector

pub fn start( envs: Box<dyn BatchSteppable>, n_steps: usize, gamma: f64, gae_lambda: f64, tx: Sender<RolloutBatch>, value_fn: Arc<dyn Fn(&[f32]) -> Vec<f64> + Send + Sync>, action_fn: Arc<dyn Fn(&[f32]) -> (Vec<f32>, Vec<f64>) + Send + Sync>, ) -> Self

Start the collector in a background thread.

The collector will repeatedly:

1. Collect n_steps of experience from all envs
2. Compute GAE advantages and returns
3. Send the resulting RolloutBatch through tx

It stops when stop() is called or the channel is disconnected.

pub fn stop(&mut self)

Signal the collector to stop and wait for the thread to finish.

pub fn is_stopped(&self) -> bool

Check whether the collector has been asked to stop.

Trait Implementations

impl Drop for AsyncCollector

fn drop(&mut self)

Executes the destructor for this type.