pim/road-gen
1
0
Fork 0
mirror of https://github.com/pim-n/road-gen.git synced 2026-02-03 01:13:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

fixes #6

Browse Source
This commit is contained in:
Pim Nelissen
2026-02-02 15:40:52 +01:00
parent 3b7c37f7c4
commit 784f22bd63
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/road_gen/generators/segmented_road_generator.py
View File

@ -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