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