def set_repeat_workflow(self, repeat_unit=None): """Set repeat workflow.""" if repeat_unit: self._root.element(tag_name="repeat-on-button").element( tag_name="a").to_subtype().click() repeat_modal = RepeatWorkflowModal() repeat_modal.set_repeat_checkbox() repeat_modal.set_repeats_select(repeat_unit) repeat_modal.click_save_and_close_btn()