浏览代码

fixing formatting

/MLA-1734-demo-provider
vincentpierre 4 年前
当前提交
52e4069f
共有 1 个文件被更改,包括 8 次插入5 次删除
  1. 13
      ml-agents/mlagents/trainers/tests/test_subprocess_env_manager.py

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


@mock.patch(
"mlagents.trainers.subprocess_env_manager.SubprocessEnvManager.create_worker"
)
def test_training_behaviors_collects_results_from_all_envs(self, mock_create_worker):
def test_training_behaviors_collects_results_from_all_envs(
self, mock_create_worker
):
worker_id, EnvironmentResponse(EnvironmentCommand.RESET, worker_id, {f"key{worker_id}" : worker_id})
worker_id,
EnvironmentResponse(
EnvironmentCommand.RESET, worker_id, {f"key{worker_id}": worker_id}
),
)
mock_create_worker.side_effect = create_worker_mock

params = {"test": "params"}
for i, env in enumerate(manager.env_workers):
for env in manager.env_workers:
@mock.patch(
"mlagents.trainers.subprocess_env_manager.SubprocessEnvManager.create_worker"

正在加载...
取消
保存