Add checks around unsafe zip operations in example_encoding tooling.
Add checks around unsafe zip operations in example_encoding tooling.

PiperOrigin-RevId: 377601685
Change-Id: I0ed38ec83375bfb909366399d8e42def557a7be2

tensorflow/agents,tensorflow/agents