Ejemplo n.º 1
0
def test_ddpg():
    args = [
        '--env-id', 'Pendulum-v0', '--num-timesteps', 1000, '--noise-type',
        'ou_0.01'
    ]
    args = list(map(str, args))
    return_code = subprocess.call(
        ['python', '-m', 'stable_baselines.ddpg.main'] + args)
    _assert_eq(return_code, 0)
Ejemplo n.º 2
0
def test_ddpg():
    args = ['--env-id', ENV_ID, '--nb-rollout-steps', 100]
    args = list(map(str, args))
    return_code = subprocess.call(['python', '-m', 'stable_baselines.ddpg.main'] + args)
    _assert_eq(return_code, 0)