25 lines
590 B
Python
Raw Normal View History

from tianshou.highlevel.env import (
EnvFactoryRegistered,
VectorEnvType,
)
class DiscreteTestEnvFactory(EnvFactoryRegistered):
def __init__(self) -> None:
super().__init__(
2024-04-24 17:06:59 +02:00
task="CartPole-v1",
train_seed=42,
test_seed=1337,
venv_type=VectorEnvType.DUMMY,
)
class ContinuousTestEnvFactory(EnvFactoryRegistered):
def __init__(self) -> None:
super().__init__(
task="Pendulum-v1",
train_seed=42,
test_seed=1337,
venv_type=VectorEnvType.DUMMY,
)