|
|
|
|
|
|
from typing import List, Tuple |
|
|
|
from mlagents_envs.base_env import ObservationSpec, DimensionProperty, SensorType |
|
|
|
from mlagents_envs.base_env import ObservationSpec, DimensionProperty, ObservationType |
|
|
|
import pytest |
|
|
|
import copy |
|
|
|
import os |
|
|
|
|
|
|
obs_specs: List[ObservationSpec] = [] |
|
|
|
for shape in shapes: |
|
|
|
dim_prop = (DimensionProperty.UNSPECIFIED,) * len(shape) |
|
|
|
spec = ObservationSpec(shape, dim_prop, SensorType.OBSERVATION) |
|
|
|
spec = ObservationSpec(shape, dim_prop, ObservationType.DEFAULT) |
|
|
|
obs_specs.append(spec) |
|
|
|
return obs_specs |