fix steps not being logged

ArchipelagoMW · Nov 12, 2023 · d8be708 · d8be708
1 parent 8800fc8
commit d8be708
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/test/benchmark/__init__.py b/test/benchmark/__init__.py
@@ -92,16 +92,15 @@ def main(self):
 
                     gc.collect()
                     for step in self.gen_steps:
-                        with TimeIt(f"{game} step {step}"):
+                        with TimeIt(f"{game} step {step}", logger):
                             call_all(multiworld, step)
                             gc.collect()
 
-                    all_state = multiworld.get_all_state(False)
                     locations = sorted(multiworld.get_unfilled_locations())
-
                     if not locations:
                         continue
 
+                    all_state = multiworld.get_all_state(False)
                     for location in locations:
                         time_taken = self.location_test(location, multiworld.state, "empty_state")
                         summary_data["empty_state"][location.name] = time_taken