def web(host, selenium_host): if selenium_host: params = { "driver_name": 'remote', "url": selenium_host, "browser": "chrome", } else: params = { "driver_name": "chrome", "executable_path": ChromeDriverManager().install() } browser = Browser(**params) browser.element_class = WebDriverElement browser.wait_time = 10 yield BeSmarterWeb(host, browser) browser.quit()