Minor changes

Electro1512 · Nov 25, 2024 · 93c51dd · 93c51dd
1 parent 08bab75
commit 93c51dd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/BlastShieldRando.py b/BlastShieldRando.py
@@ -213,8 +213,8 @@ def _get_available_blast_shields(
 
 def apply_blast_shield_mapping(world: "MetroidPrimeWorld"):
     remove_vanilla_blast_shields(world)
+    assert world.blast_shield_mapping is not None
     mapping = world.blast_shield_mapping
-    assert mapping is not None
     for area, area_mapping in mapping.items():
         for room_name, door_mapping in area_mapping.type_mapping.items():
             for door_id, shield_type in door_mapping.items():

diff --git a/DoorRando.py b/DoorRando.py
@@ -156,6 +156,7 @@ def remap_doors_to_power_beam_if_necessary(world: "MetroidPrimeWorld"):
         starting_beam = world.starting_room_data.selected_loadout.starting_beam
 
         if starting_beam is not SuitUpgrade.Power_Beam:
+            assert world.starting_room_data
             for area, mapping in world.door_color_mapping.items():
                 if (
                     area == world.starting_room_data.area.value