浏览代码

Fix subprocess test

/develop/sac-apex
Ervin Teng 5 年前
当前提交
441fbb91
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/tests/test_subprocess_env_manager.py

2
ml-agents/mlagents/trainers/tests/test_subprocess_env_manager.py


# Test policy queue
mock_policy = mock.Mock()
agent_manager_mock.policy_queue.get_nowait.return_value = mock_policy
agent_manager_mock.policy_queue.get.return_value = mock_policy
env_manager.advance()
assert env_manager.policies[brain_name] == mock_policy
assert agent_manager_mock.policy == mock_policy

正在加载...
取消
保存