fix profiling bug

lavague-ai · Sep 6, 2024 · 052c363 · 052c363
1 parent b343013
commit 052c363
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lavague-core/lavague/core/utilities/profiling.py b/lavague-core/lavague/core/utilities/profiling.py
@@ -61,6 +61,8 @@ def time_profiler(
         if full_step_profiling:
             agent_steps.append(record)
         else:
+            if len(agent_events) == 0:
+                start_new_step()
             agent_events[-1].append(record)