浏览代码

Fix PPO test

/release-0.13.1
Ervin Teng 5 年前
当前提交
e83276f6
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1
      ml-agents/mlagents/trainers/tests/test_ppo.py

1
ml-agents/mlagents/trainers/tests/test_ppo.py


brain_params.brain_name, 0, trainer_params, True, False, 0, "0", False
)
policy_mock = mock.Mock(spec=PPOPolicy)
policy_mock.get_current_step.return_value = 0
step_count = (
5
) # 10 hacked because this function is no longer called through trainer

正在加载...
取消
保存