Change trajectory writer API - replace get_signature_timeout_ms with retrieve_signature. Setting correct value of get_signature_timeout_ms is problematic, while there are 2 real use cases - whether to get signature, or not.
Change trajectory writer API - replace get_signature_timeout_ms with retrieve_signature. Setting correct value of get_signature_timeout_ms is problematic, while there are 2 real use cases - whether to get signature, or not.

PiperOrigin-RevId: 442166767
Change-Id: Ic0070aec612b40a060a4540bc3580b0cb54731d0

tensorflow/agents,tensorflow/agents