示例#1
0
文件： conftest.py 项目： c-w-m/slm-lab 
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
示例#2
0
文件： conftest.py 项目： c-w-m/slm-lab 
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
示例#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