Skip to content

lit_sb3.wrappers.yaml_compat_resize

[docs] module lit_sb3.wrappers.yaml_compat_resize 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
from __future__ import annotations

import gymnasium as gym
from gymnasium.wrappers import ResizeObservation


class YAMLCompatResizeObservation(ResizeObservation):
    """ResizeObservation that accepts a list instead of a tuple for YAML compatibility."""

    def __init__(self, env: gym.Env, shape: list[int]) -> None:
        super().__init__(env, (shape[0], shape[1]))