Firstly you should import this library. (from selenium.webdriver.common.by import By) Then, you can create an assignment called button. After call the function. Like that;
button = driver.find_element(By.ID, "button-id")
button.click()
You can set it with your own code.