border-py-gym-env 0.0.5

Reinforcement learning library
//! Parameters of atari environments
use serde::{Serialize, Deserialize};

#[derive(Debug, Serialize, Deserialize)]
/// Specifies training or evaluation mode.
#[derive(Clone)]
pub enum AtariWrapper {
    /// Training mode
    Train,

    /// Evaluation mode
    Eval,
}