def find_class_with_timeout(driver, CLASS, timeout=TIMEOUT): return WebDriverWait(driver, timeout).until( EC.presence_of_element_located((By.CLASS_NAME, CLASS)) )