浏览代码

Fix subprocess env manager test

/develop/sac-apex
Ervin Teng 5 年前
当前提交
370b3c40
共有 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


external_brains_mock.return_value = [brain_name]
agent_manager_mock = mock.Mock()
mock_policy = mock.Mock()
agent_manager_mock.policy_queue.get.side_effect = [
agent_manager_mock.policy_queue.get_nowait.side_effect = [
mock_policy,
mock_policy,
AgentManagerQueue.Empty(),

正在加载...
取消
保存