import numpy as onp
import pytest
from dm_control import suite


def flat_obs(o):
    return onp.concatenate([o[k].flatten() for k in o])


@pytest.fixture
def cartpole_environment(seed: int = 42):
    env = suite.load("cartpole", "swingup", {"random": seed})
    return env

8e31822ebd5b8ecfb1929a7f202147a9e807e2d9
<mask token>

<mask token>


def flat_obs(o):
    return onp.concatenate([o[k].flatten() for k in o])


<mask token>

<mask token>


def flat_obs(o):
    return onp.concatenate([o[k].flatten() for k in o])


@pytest.fixture
def cartpole_environment(seed: int=42):
    env = suite.load('cartpole', 'swingup', {'random': seed})
    return env

import numpy as onp
import pytest
from dm_control import suite


def flat_obs(o):
    return onp.concatenate([o[k].flatten() for k in o])


@pytest.fixture
def cartpole_environment(seed: int=42):
    env = suite.load('cartpole', 'swingup', {'random': seed})
    return env

import numpy as onp
import pytest
from dm_control import suite


def flat_obs(o):
    return onp.concatenate([o[k].flatten() for k in o])


@pytest.fixture
def cartpole_environment(seed: int = 42):
    env = suite.load("cartpole", "swingup", {"random": seed})
    return env
