Passing policy_state correctly into policy.action() function when computing Q values.
Passing policy_state correctly into policy.action() function when computing Q values.

PiperOrigin-RevId: 402601593
Change-Id: Ibb58d385835ccf9a1062be8a7a281a7f4dd92bdc

tensorflow/agents,tensorflow/agents