Fix a bug in Neural LinUCB where we neglected to update the bool variable on whether to use reward layer or lin ucb for inference in the policy.
Fix a bug in Neural LinUCB where we neglected to update the bool variable on whether to use reward layer or lin ucb for inference in the policy.

PiperOrigin-RevId: 279993500
Change-Id: Icf564f637d27b32c43891023a9c37db3f52c3973

tensorflow/agents,tensorflow/agents