Skip to content

Commit 1601324

Browse files
committed
Fix
1 parent 9635880 commit 1601324

File tree

5 files changed

+7
-10
lines changed

5 files changed

+7
-10
lines changed

car_dreamer/carla_navigation_env.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ class CarlaNavigationEnv(CarlaWptEnv):
1313
* ``num_vehicles``: Number of vehicles to spawn in the environment
1414
1515
"""
16-
16+
1717
SUCCESS_DISTANCE_THRESHOLD = 200.0
1818

1919
def on_reset(self) -> None:
@@ -23,7 +23,7 @@ def on_reset(self) -> None:
2323
self.waypoints, self.planner_stats = self.ego_planner.run_step()
2424
self.num_completed = self.planner_stats["num_completed"]
2525
self.sum_travel_distance = self.planner_stats["travel_distance"]
26-
26+
2727
def on_step(self) -> None:
2828
super().on_step()
2929
self.sum_travel_distance += self.planner_stats["travel_distance"]

car_dreamer/carla_right_turn_random_env.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,9 @@
1-
import random
2-
import time
3-
from collections import deque
1+
import numpy as np
42

53
import carla
64

75
from .carla_wpt_fixed_env import CarlaWptFixedEnv
8-
from .toolkit import FixedPathPlanner, get_vehicle_pos
6+
from .toolkit import FixedPathPlanner
97

108

119
class CarlaRightTurnRandomEnv(CarlaWptFixedEnv):
@@ -19,8 +17,7 @@ def __init__(self, config):
1917
super().__init__(config)
2018

2119
def on_reset(self) -> None:
22-
random.seed(time.time())
23-
random_index = random.randint(0, len(self._config.lane_start_point))
20+
random_index = np.random.randint(0, len(self._config.lane_start_point))
2421
self.ego_src = self._config.lane_start_point[random_index]
2522
ego_transform = carla.Transform(carla.Location(*self.ego_src[:3]), carla.Rotation(yaw=self.ego_src[3]))
2623
self.ego = self._world.spawn_actor(transform=ego_transform)

car_dreamer/carla_stop_sign_env.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ def __init__(self, config):
1818
super().__init__(config)
1919

2020
def on_reset(self) -> None:
21-
np.random.seed(time.time())
2221
random_index = np.random.randint(0, len(self._config.lane_start_point))
2322
self.ego_src = self._config.lane_start_point[random_index]
2423
ego_transform = carla.Transform(carla.Location(*self.ego_src[:3]), carla.Rotation(yaw=self.ego_src[3]))

car_dreamer/carla_traffic_lights_env.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ def __init__(self, config):
1818
super().__init__(config)
1919

2020
def on_reset(self) -> None:
21-
np.random.seed(time.time())
2221
random_index = np.random.randint(0, len(self._config.lane_start_point))
2322
self.ego_src = self._config.lane_start_point[random_index]
2423
ego_transform = carla.Transform(carla.Location(*self.ego_src[:3]), carla.Rotation(yaw=self.ego_src[3]))

car_dreamer/toolkit/observer/handlers/renderer/birdeye_renderer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -313,6 +313,8 @@ def _render_traffic_light(self, surface, traffic_light: carla.Actor, color: Colo
313313

314314
def _render_stop_signs(self, **env_state):
315315
stop_sign_state = env_state.get("stop_sign_state")
316+
if stop_sign_state is None:
317+
return
316318
stop_signs = self._world_manager.carla_actors("stop")
317319
for stop_sign in stop_signs:
318320
if stop_sign.id in stop_sign_state:

0 commit comments

Comments
 (0)