From 784f22bd63c322a7a8d30151ee2c66e5bcc56e90 Mon Sep 17 00:00:00 2001 From: Pim Nelissen Date: Mon, 2 Feb 2026 15:40:52 +0100 Subject: [PATCH] fixes #6 --- src/road_gen/generators/segmented_road_generator.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/road_gen/generators/segmented_road_generator.py b/src/road_gen/generators/segmented_road_generator.py index dde5c6f..959022f 100644 --- a/src/road_gen/generators/segmented_road_generator.py +++ b/src/road_gen/generators/segmented_road_generator.py @@ -47,8 +47,9 @@ class SegmentedRoadGenerator(BaseRoadGenerator): Returns: Tuple[np.ndarray, np.ndarray]: x and y coordinates of the waypoints describing the random road. """ - if not all(segment in prefabs.PREFABS.keys() for segment in segments): - raise ValueError(f"Invalid segment type provided. Available choices: {prefabs.SEGMENTS.keys()}") + existing_prefabs = prefabs.PREFABS.keys() + if not all(segment in existing_prefabs for segment in segments): + raise ValueError(f"Invalid segment type provided. Available choices: {existing_prefabs}") self.segments = segments self.alpha = alpha