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