Allow arbitrary Sample type in the agent builder.
Allow arbitrary Sample type in the agent builder.

PiperOrigin-RevId: 382940625
Change-Id: I3f7de5c6c068ac181ab76266363cd33d4b283326

deepmind/acme,deepmind/acme