Use TrajectoryWriter in demo colab.
Use TrajectoryWriter in demo colab.

PiperOrigin-RevId: 358159026
Change-Id: I581f63b3aced52f5f9a9b7ab84e6c5c198b649da

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