Add option for datasets matching reverb trajectory APIs in the learner.
Add option for datasets matching reverb trajectory APIs in the learner.

PiperOrigin-RevId: 390740059
Change-Id: I52be5e6da6895236548671cf9e1de8820939234d

tensorflow/agents,tensorflow/agents