浏览代码

Fix tests

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

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


new_step_info.current_all_brain_info[brain_name],
new_step_info.brain_name_to_action_info[brain_name].outputs,
)
trainer_mock.clear_update_buffer.assert_called_once()
trainer_mock.advance.assert_called_once()

1
ml-agents/mlagents/trainers/trainer_controller.py


trainer.parameters["time_horizon"]
if "time_horizon" in trainer.parameters
else None,
trainer.summary_path,
)
)
self.managers[name] = agent_manager

正在加载...
取消
保存