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

Merge pull request #11 from pim-n/fix-prefab-bug

Browse Source 
fixes #6
This commit is contained in:
Pim Nelissen
2026-02-02 15:43:11 +01:00
committed by GitHub
parent 3b7c37f7c4 784f22bd63
commit 36f4abb425
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