浏览代码

fix reward provider tests

/develop/action-spec-gym
Andrew Cohen 4 年前
当前提交
5f0f7e3a
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 4
      ml-agents/mlagents/trainers/tests/torch/test_reward_providers/utils.py

4
ml-agents/mlagents/trainers/tests/torch/test_reward_providers/utils.py


]
action_buffer = behavior_spec.action_spec.random_action(1)
action = {}
if action_buffer.continuous is not None:
if behavior_spec.action_spec.continuous_size > 0:
if action_buffer.discrete is not None:
if behavior_spec.action_spec.discrete_size > 0:
action["discrete_action"] = action_buffer.discrete
for _ in range(number):

正在加载...
取消
保存