Sample action from an action space for given state

def step(self, observations):
        """ Sample action from an action space for given state """
        log_histogram = self(observations)
        actions = self.q_head.sample(log_histogram)

        return {
            'actions': actions,
            'log_histogram': log_histogram
        }