80d314ae5eaadc88061a62c827e1b8670098c1ab,rllib/utils/exploration/random.py,Random,get_tf_exploration_action_op,#Random#Any#Any#,56

Before Change

    @tf_function(tf)
    def get_tf_exploration_action_op(self, action_dist, explore):
        if explore:
            action = tf.py_function(self.action_space.sample, [],
                                    self.dtype_sample)
            // Will be unnecessary, once we support batch/time-aware Spaces.
            action = tf.expand_dims(tf.cast(action, dtype=self.dtype), 0)
        else:
            action = tf.cast(
                action_dist.deterministic_sample(), dtype=self.dtype)

        // TODO(sven): Move into (deterministic_)sample(logp=True|False)
        if isinstance(action, TupleActions):
            batch_size = tf.shape(action[0][0])[0]
        else:
            batch_size = tf.shape(action)[0]

After Change

            return tf.cast(
                action_dist.deterministic_sample(), dtype=self.dtype)

        action = tf.cond(
            pred=tf.constant(explore, dtype=tf.bool)
            if isinstance(explore, bool) else explore,
            true_fn=true_fn,
            false_fn=false_fn)

        // TODO(sven): Move into (deterministic_)sample(logp=True|False)
        if isinstance(action, TupleActions):
            batch_size = tf.shape(action[0][0])[0]

Instances