Skip to content

Commit f95ab67

Browse files
committed
Fix missing parameter and argument.
1 parent b0cb2ab commit f95ab67

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

smarts/env/utils/observation_conversion.py

+10-3
Original file line numberDiff line numberDiff line change
@@ -765,7 +765,9 @@ def name(self):
765765

766766

767767
neighborhood_vehicle_states_space_format = StandardConfigurableSpaceFormat(
768-
lambda obs: _format_neighborhood_vehicle_states(obs.neighborhood_vehicle_states),
768+
lambda obs, agent_interface: _format_neighborhood_vehicle_states(
769+
obs.neighborhood_vehicle_states, agent_interface
770+
),
769771
lambda agent_interface: bool(agent_interface.neighborhood_vehicle_states),
770772
"neighborhood_vehicle_states",
771773
_configure_neighborhood_vehicle_states_space,
@@ -1072,7 +1074,7 @@ class ObservationSpacesFormatter:
10721074
vehicles are insufficient, default feature values are padded.
10731075
"neighborhood_vehicle_states": dict({
10741076
"box":
1075-
Bounding box of neighbor vehicles. Defaults to np.array([0,0,0]) per
1077+
Bounding box of neighbor vehicles. Defaults to np.array([.0,.0,.0]) per
10761078
vehicle. shape=(10,3). dtype=np.float32.
10771079
"heading":
10781080
Heading of neighbor vehicles in radians [-pi, pi]. Defaults to
@@ -1091,11 +1093,16 @@ class ObservationSpacesFormatter:
10911093
vehicle. shape=(10,). dtype=np.int8.
10921094
"position":
10931095
Coordinate of the center of neighbor vehicles' bounding box's bottom
1094-
plane. Defaults to np.array([0,0,0]) per vehicle. shape=(10,3).
1096+
plane. Defaults to np.array([.0,.0,.0]) per vehicle. shape=(10,3).
10951097
dtype=np.float64.
10961098
"speed":
10971099
Speed of neighbor vehicles in m/s. Defaults to np.array([0]) per
10981100
vehicle. shape=(10,). dtype=np.float32.
1101+
"lane_position":
1102+
A reference line coordinates. Coordinates are s, t, and h relating
1103+
to offset along lane, horizontal displacement, and surface
1104+
displacement. Defaults to np.array([.0,.0,.0]) per vehicle. shape=(10,3).
1105+
dtype=np.float64
10991106
})
11001107
11011108
Occupancy grid map. Map is binary, with 255 if a cell is occupied, else 0.

0 commit comments

Comments
 (0)