def get_element_click(self, state):
"""Get the action that clicks the button."""
for element in state.dom_elements:
if element.tag == 'button':
action = MiniWoBElementClick(element, fail_hard=True)
print('Clicking with {}'.format(action))
return action
raise ValueError('Cannot find button: {}'.format(str(state.dom_elements)))
def get_input_click(self, state):
"""Get the action that clicks an input element."""
for element in state.dom_elements:
if element.tag == 'input_text':
action = MiniWoBElementClick(element, fail_hard=True)
print 'Clicking with {}'.format(action)
return action
raise ValueError('Cannot find input: {}'.format(str(
state.dom_elements)))
def create_element_click_action(self, element):
action = MiniWoBElementClick(element, fail_hard=True)
print 'Clicking with {}'.format(action)
return action