vizdoom-turbo 0.1.2

Native-vector Gymnasium environments for high-throughput ViZDoom rollouts

⚡ High-throughput ViZDoom rollouts, native-vector by design. ⚡

vizdoom-turbo is a Python library for reinforcement-learning researchers who need fast, parallel ViZDoom environments. It provides a Gymnasium vector environment that can be used directly or selected as an isolated environment provider in rlab.

Each vector lane owns an independent DoomGame. Lanes advance concurrently through ViZDoom's native API, while Rust processes frames in batches for policy-ready observations.

Install

git clone [e-mail]:tsilva/ViZDoom-turbo-fork.git
cd ViZDoom-turbo-fork/turbo
uv sync --all-extras

Run Python and project commands through uv run.

Use

import numpy as np
from vizdoom_turbo import VizdoomTurboVecEnv

env = VizdoomTurboVecEnv(
    "VizdoomBasic-v1",
    num_envs=16,
    num_threads=8,
    obs_resize=(84, 84),
    obs_grayscale=True,
    obs_layout="chw",
    frame_skip=4,
    frame_stack=4,
    use_restricted_actions="minimal",
)

try:
    observations, infos = env.reset(seed=7)
    actions = np.zeros(env.num_envs, dtype=np.int64)
    observations, rewards, terminated, truncated, infos = env.step(actions)

    done = terminated | truncated
    if np.any(done):
        observations, infos = env.reset(
            options={
                "reset_mask": done,
                "state_indices": np.zeros(env.num_envs, dtype=np.int32),
            }
        )
finally:
    env.close()

The package accepts registered Vizdoom... Gymnasium IDs and ViZDoom .cfg paths. It also registers ...-Turbo-v0 vector aliases for the built-in visual scenarios.

Use with rlab

Install this distribution in the rlab runtime, then select its provider:

environment:
  env_provider: vizdoom-turbo
  env_config:
    game: VizdoomBasic-v1
    state: default
    n_envs: 16
    env_args:
      num_threads: 8
      use_restricted_actions: minimal
      obs_grayscale: true
      obs_layout: chw
      frame_stack: 4
    preprocessing:
      frame_skip: 4
      max_pool_frames: true
      observation_size: 84
      obs_resize_algorithm: area
    task:
      id: identity
      action: {set: native}
      signals: {}
      events: {}
      termination: {}
      reward: {reward_mode: native}

Commands

uv sync --all-extras                                      # install project and dev dependencies
uv run pytest -q                                          # run Python and live-environment tests
uv run ruff check .                                       # lint Python
cargo fmt --check                                         # check Rust formatting
cargo clippy --all-targets --all-features -- -D warnings  # lint Rust
uv build --wheel                                          # build the distributable wheel

Notes

• Python 3.11–3.14 is supported. Source builds require Rust 1.85 or newer.
• ViZDoom 1.3.0 supplies built-in scenarios and Freedoom assets. Commercial Doom IWADs are not included; pass one with rom_path when required.
• Autoreset is disabled. Terminal lanes retain their final observation and must be selected explicitly with a masked reset.
• Preprocessing supports crop removal or masking, max-pooling, nearest/bilinear/area resize, grayscale or RGB, frame skip, frame stacking, and CHW or HWC layouts.
• The native vector path supports image observations, rgb_array rendering, and one player. Recording is not supported.

Architecture

License

MIT