tools/rerun: correct reset to defined default blueprint (#32692)

tools/rerun/run.py

@@ -53,9 +53,9 @@ def log_thumbnail(thumbnailMsg):
   rr.log("/thumbnail", rr.ImageEncoded(contents=bytesImgData))
 
 @rr.shutdown_at_exit
-def process(lr):
+def process(blueprint, lr):
   rr.init("rerun_test")
-  rr.connect()
+  rr.connect(default_blueprint=blueprint)
 
   ret = []
   for msg in lr:
@@ -80,10 +80,10 @@ def process(lr):
   args = parser.parse_args()
 
   blueprint = createBlueprint()
-  rr.init("rerun_test", default_blueprint=blueprint)
+  rr.init("rerun_test")
   rr.spawn(connect=False) # child processes stream data to Viewer
 
   route_or_segment_name = DEMO_ROUTE if args.demo else args.route_or_segment_name.strip()
   print("Getting route log paths")
   lr = LogReader(route_or_segment_name)
-  lr.run_across_segments(NUM_CPUS, partial(process))
+  lr.run_across_segments(NUM_CPUS, partial(process, blueprint))