code: def _get_tf_exploration_action(self, action_dist, explore):
      action = tf.argmax(
          tf.cond(
              pred=explore,
              true_fn=lambda: action_dist.inputs,
              false_fn=lambda: self.model.predict(
                  self.model.current_obs()
              ),
          ),
          axis=-1,
      )
      return action, None
  
file_path: ray/rllib/utils/exploration/thompson_sampling.py
imports: from typing import Union
  from ray.rllib.models.action_dist import ActionDistribution
  from ray.rllib.utils.annotations import override, PublicAPI
  from ray.rllib.utils.exploration.exploration import Exploration
  from ray.rllib.utils.framework import TensorType, try_import_tf
label_desc: Use image node_id 3 for calling the ThompsonSampling obj's underlying member method code with example usage: obj._get_tf_exploration_action(action_dist, explore) and returns: action,  None