mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-07-30 21:34:30 +00:00
Rename RenderObservation
wrapper to AddRenderObservation
(#904)
This commit is contained in:
@@ -18,7 +18,7 @@
|
||||
.. autoclass:: gymnasium.wrappers.GrayscaleObservation
|
||||
.. autoclass:: gymnasium.wrappers.MaxAndSkipObservation
|
||||
.. autoclass:: gymnasium.wrappers.NormalizeObservation
|
||||
.. autoclass:: gymnasium.wrappers.RenderObservation
|
||||
.. autoclass:: gymnasium.wrappers.AddRenderObservation
|
||||
.. autoclass:: gymnasium.wrappers.ResizeObservation
|
||||
.. autoclass:: gymnasium.wrappers.ReshapeObservation
|
||||
.. autoclass:: gymnasium.wrappers.RescaleObservation
|
||||
|
@@ -56,7 +56,7 @@ wrapper in the page on the wrapper type
|
||||
- Records videos of environment episodes using the environment's render function.
|
||||
* - :class:`RenderCollection`
|
||||
- Collect rendered frames of an environment such ``render`` returns a ``list[RenderedFrame]``.
|
||||
* - :class:`RenderObservation`
|
||||
* - :class:`AddRenderObservation`
|
||||
- Includes the rendered observations in the environment's observations.
|
||||
* - :class:`RescaleAction`
|
||||
- Affinely (linearly) rescales a ``Box`` action space of the environment to within the range of ``[min_action, max_action]``.
|
||||
|
@@ -73,11 +73,11 @@ from gymnasium.wrappers.transform_action import (
|
||||
TransformAction,
|
||||
)
|
||||
from gymnasium.wrappers.transform_observation import (
|
||||
AddRenderObservation,
|
||||
DtypeObservation,
|
||||
FilterObservation,
|
||||
FlattenObservation,
|
||||
GrayscaleObservation,
|
||||
RenderObservation,
|
||||
RescaleObservation,
|
||||
ReshapeObservation,
|
||||
ResizeObservation,
|
||||
@@ -99,7 +99,7 @@ __all__ = [
|
||||
"TransformObservation",
|
||||
"MaxAndSkipObservation",
|
||||
"NormalizeObservation",
|
||||
"RenderObservation",
|
||||
"AddRenderObservation",
|
||||
"ResizeObservation",
|
||||
"ReshapeObservation",
|
||||
"RescaleObservation",
|
||||
@@ -142,7 +142,7 @@ _wrapper_to_class = {
|
||||
_renamed_wrapper = {
|
||||
"AutoResetWrapper": "Autoreset",
|
||||
"FrameStack": "FrameStackObservation",
|
||||
"PixelObservationWrapper": "RenderObservation",
|
||||
"PixelObservationWrapper": "AddRenderObservation",
|
||||
"VectorListInfo": "vector.DictInfoToList",
|
||||
}
|
||||
|
||||
|
@@ -31,7 +31,7 @@ __all__ = [
|
||||
"ReshapeObservation",
|
||||
"RescaleObservation",
|
||||
"DtypeObservation",
|
||||
"RenderObservation",
|
||||
"AddRenderObservation",
|
||||
]
|
||||
|
||||
|
||||
@@ -607,7 +607,7 @@ class DtypeObservation(
|
||||
)
|
||||
|
||||
|
||||
class RenderObservation(
|
||||
class AddRenderObservation(
|
||||
TransformObservation[WrapperObsType, ActType, ObsType],
|
||||
gym.utils.RecordConstructorArgs,
|
||||
):
|
||||
@@ -620,7 +620,7 @@ class RenderObservation(
|
||||
|
||||
Example - Replace the observation with the rendered image:
|
||||
>>> env = gym.make("CartPole-v1", render_mode="rgb_array")
|
||||
>>> env = RenderObservation(env, render_only=True)
|
||||
>>> env = AddRenderObservation(env, render_only=True)
|
||||
>>> env.observation_space
|
||||
Box(0, 255, (400, 600, 3), uint8)
|
||||
>>> obs, _ = env.reset(seed=123)
|
||||
@@ -634,7 +634,7 @@ class RenderObservation(
|
||||
|
||||
Example - Add the rendered image to the original observation as a dictionary item:
|
||||
>>> env = gym.make("CartPole-v1", render_mode="rgb_array")
|
||||
>>> env = RenderObservation(env, render_only=False)
|
||||
>>> env = AddRenderObservation(env, render_only=False)
|
||||
>>> env.observation_space
|
||||
Dict('pixels': Box(0, 255, (400, 600, 3), uint8), 'state': Box([-4.8000002e+00 -3.4028235e+38 -4.1887903e-01 -3.4028235e+38], [4.8000002e+00 3.4028235e+38 4.1887903e-01 3.4028235e+38], (4,), float32))
|
||||
>>> obs, info = env.reset(seed=123)
|
||||
@@ -651,7 +651,7 @@ class RenderObservation(
|
||||
|
||||
Change logs:
|
||||
* v0.15.0 - Initially added as ``PixelObservationWrapper``
|
||||
* v1.0.0 - Renamed to ``RenderObservation``
|
||||
* v1.0.0 - Renamed to ``AddRenderObservation``
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
@@ -661,7 +661,7 @@ class RenderObservation(
|
||||
render_key: str = "pixels",
|
||||
obs_key: str = "state",
|
||||
):
|
||||
"""Constructor of the pixel observation wrapper.
|
||||
"""Constructor of the add render observation wrapper.
|
||||
|
||||
Args:
|
||||
env: The environment to wrap.
|
||||
|
@@ -3,7 +3,7 @@ import numpy as np
|
||||
import pytest
|
||||
|
||||
from gymnasium import spaces
|
||||
from gymnasium.wrappers import RenderObservation
|
||||
from gymnasium.wrappers import AddRenderObservation
|
||||
from tests.testing_env import GenericTestEnv
|
||||
|
||||
|
||||
@@ -30,7 +30,7 @@ def test_dict_observation(pixels_only, pixel_key="rgb"):
|
||||
# width, height = (320, 240)
|
||||
|
||||
# The wrapper should only add one observation.
|
||||
wrapped_env = RenderObservation(
|
||||
wrapped_env = AddRenderObservation(
|
||||
env,
|
||||
render_key=pixel_key,
|
||||
render_only=pixels_only,
|
||||
@@ -69,7 +69,7 @@ def test_single_array_observation(pixels_only):
|
||||
assert isinstance(env.observation_space, spaces.Box)
|
||||
|
||||
# The wrapper should only add one observation.
|
||||
wrapped_env = RenderObservation(
|
||||
wrapped_env = AddRenderObservation(
|
||||
env,
|
||||
render_key=pixel_key,
|
||||
render_only=pixels_only,
|
Reference in New Issue
Block a user