Python make_agent_envの例

コード例 #1

ファイル: 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

ファイル: 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

ファイル: conftest.py プロジェクト: wantajob/SLM-Lab

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