Swap to explicit key error

This commit is contained in:
Justus Lind
2024-11-21 00:20:35 +10:00
parent 9fe4f221cf
commit 719255891e

View File

@@ -183,8 +183,10 @@ class MuseDashWorld(World):
if album:
return MuseDashSongItem(name, self.player, album)
song = self.md_collection.song_items[name]
return MuseDashSongItem(name, self.player, song)
song = self.md_collection.song_items.get(name)
if song:
return MuseDashSongItem(name, self.player, song)
raise KeyError(name)
def get_filler_item_name(self) -> str:
return self.random.choices(self.filler_item_names, self.filler_item_weights)[0]