Esempio n. 1
0
def test_prioritized_replay_memory(request):
    spec = spec_util.get('experimental/misc/base.json',
                         'base_prioritized_replay_memory')
    spec_util.tick(spec, 'trial')
    agent, env = make_agent_env(spec)
    res = (agent.body.memory, ) + request.param
    return res
Esempio n. 2
0
def test_on_policy_batch_memory(request):
    spec = spec_util.get('experimental/misc/base.json',
                         'base_on_policy_batch_memory')
    spec_util.tick(spec, 'trial')
    agent, env = make_agent_env(spec)
    res = (agent.body.memory, ) + request.param
    return res
Esempio n. 3
0
def test_memory(request):
    spec = spec_util.get('experimental/misc/base.json', 'base_memory')
    agent, env = make_agent_env(spec)
    res = (agent.body.memory, ) + request.param
    return res