Expand description
Prelude module for convenient imports.
Brings the highest-value public types into scope with a single
use thrust_rl::prelude::*;. The set is curated, not a blanket glob:
- Environment API (always available): the
Environmenttrait plusStepResult,SpaceInfo,SpaceType, and the built-inCartPole/SimpleBanditenvironments. - Training stack (requires the
trainingfeature): theMlpBurnPolicyactor-critic network and itsMlpBurnConfig, the trainer configs/trainers (A2C, PPO, DQN, SAC, BC), and theRolloutBuffer/ReplayBufferexperience buffers, and theEnvPoolvectorized env wrapper.
Training-only re-exports are #[cfg(feature = "training")]-gated so
--no-default-features still compiles.
Re-exportsยง
pub use crate::buffer::replay::ReplayBuffer;pub use crate::buffer::rollout::RolloutBuffer;pub use crate::env::pool::EnvPool;pub use crate::env::CartPole;pub use crate::env::Environment;pub use crate::env::SimpleBandit;pub use crate::env::SpaceInfo;pub use crate::env::SpaceType;pub use crate::env::StepResult;pub use crate::policy::mlp::BurnActivation;pub use crate::policy::mlp::MlpBurnConfig;pub use crate::policy::mlp::MlpBurnPolicy;pub use crate::train::A2cConfig;pub use crate::train::A2cTrainer;pub use crate::train::BcConfig;pub use crate::train::BcTrainer;pub use crate::train::DQNConfig;pub use crate::train::DQNTrainerBurn;pub use crate::train::PPOConfig;pub use crate::train::PPOTrainerBurn;pub use crate::train::SacConfig;pub use crate::train::SacTrainer;