6cbee188b8
of number of environments in SamplingConfig is used (values are now passed to factory method) This is clearer and removes the need to pass otherwise unnecessary configuration to environment factories at construction
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
from test.highlevel.env_factory import ContinuousTestEnvFactory
|
|
|
|
import pytest
|
|
|
|
from tianshou.highlevel.config import SamplingConfig
|
|
from tianshou.highlevel.experiment import (
|
|
A2CExperimentBuilder,
|
|
DDPGExperimentBuilder,
|
|
ExperimentConfig,
|
|
PGExperimentBuilder,
|
|
PPOExperimentBuilder,
|
|
REDQExperimentBuilder,
|
|
SACExperimentBuilder,
|
|
TD3ExperimentBuilder,
|
|
TRPOExperimentBuilder,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"builder_cls",
|
|
[
|
|
PPOExperimentBuilder,
|
|
A2CExperimentBuilder,
|
|
SACExperimentBuilder,
|
|
DDPGExperimentBuilder,
|
|
TD3ExperimentBuilder,
|
|
# NPGExperimentBuilder, # TODO test fails non-deterministically
|
|
REDQExperimentBuilder,
|
|
TRPOExperimentBuilder,
|
|
PGExperimentBuilder,
|
|
],
|
|
)
|
|
def test_experiment_builder_continuous_default_params(builder_cls):
|
|
env_factory = ContinuousTestEnvFactory()
|
|
sampling_config = SamplingConfig(
|
|
num_epochs=1,
|
|
step_per_epoch=100,
|
|
num_train_envs=2,
|
|
num_test_envs=2,
|
|
)
|
|
experiment_config = ExperimentConfig(persistence_enabled=False)
|
|
builder = builder_cls(
|
|
experiment_config=experiment_config,
|
|
env_factory=env_factory,
|
|
sampling_config=sampling_config,
|
|
)
|
|
experiment = builder.build()
|
|
experiment.run("test")
|
|
print(experiment)
|