code: def step(self, observations):
          
          log_histogram = self(observations)
          actions = self.q_head.sample(log_histogram)
  
          return {
              'actions': actions,
              'log_histogram': log_histogram
          }
docs: Sample action from an action space for given state