fix: flaky test caused by compare stats
@@ -123,6 +123,7 @@ def compare_stats() -> None:
instance_from_api_map[r["username"]] += r["seconds"]
for ins in instance_from_api_map:
# make sure instance initialization time is less than 5 mins
+ if ins in gpu_from_api_map:
assert instance_from_api_map[ins] - gpu_from_api_map[ins] < 60 * 5
print(f"Agent time: logs={gpu_from_log}, api={gpu_from_api}")
