29 lines
821 B
Python
29 lines
821 B
Python
|
from test.highlevel.env_factory import DiscreteTestEnvFactory
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from tianshou.highlevel.config import RLSamplingConfig
|
||
|
|
from tianshou.highlevel.experiment import (
|
||
|
|
A2CExperimentBuilder,
|
||
|
|
DQNExperimentBuilder,
|
||
|
|
PPOExperimentBuilder,
|
||
|
|
RLExperimentConfig,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"builder_cls",
|
||
|
|
[PPOExperimentBuilder, A2CExperimentBuilder, DQNExperimentBuilder],
|
||
|
|
)
|
||
|
|
def test_experiment_builder_discrete_default_params(builder_cls):
|
||
|
|
env_factory = DiscreteTestEnvFactory()
|
||
|
|
sampling_config = RLSamplingConfig(num_epochs=1, step_per_epoch=100)
|
||
|
|
builder = builder_cls(
|
||
|
|
experiment_config=RLExperimentConfig(),
|
||
|
|
env_factory=env_factory,
|
||
|
|
sampling_config=sampling_config,
|
||
|
|
)
|
||
|
|
experiment = builder.build()
|
||
|
|
experiment.run("test")
|
||
|
|
print(experiment)
|