Since actions can now be np.array, "if action" is ambiguous and crashes.
Since actions can now be np.array, "if action" is ambiguous and crashes.

PiperOrigin-RevId: 269817463

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor