Fix data race in trajectory_writer_test.cc.
Fix data race in trajectory_writer_test.cc.

PiperOrigin-RevId: 354268397
Change-Id: I329f1c5b426bb391464797956abc26c0c5dee05d

deepmind/reverb,deepmind/reverb,deepmind/reverb,deepmind/reverb