浏览代码

Fix gym tests

/MLA-1734-demo-provider
Arthur Juliani 4 年前
当前提交
097c8fa5
共有 1 个文件被更改,包括 3 次插入1 次删除
  1. 4
      gym-unity/gym_unity/tests/test_gym.py

4
gym-unity/gym_unity/tests/test_gym.py


from mlagents_envs.base_env import (
BehaviorSpec,
ActionSpec,
SensorType,
DecisionSteps,
TerminalSteps,
BehaviorMapping,

obs_shapes = [(vector_observation_space_size,)]
for _ in range(number_visual_observations):
obs_shapes += [(8, 8, 3)]
return BehaviorSpec(obs_shapes, action_spec)
sensor_types = [SensorType.OBSERVATION for _ in range(len(obs_shapes))]
return BehaviorSpec(obs_shapes, sensor_types, action_spec)
def create_mock_vector_steps(specs, num_agents=1, number_visual_observations=0):

正在加载...
取消
保存