results = {} for driver_name in ["chrome", "firefox"]: if driver_name == "firefox": from selenium.webdriver import Firefox as Driver from selenium.webdriver.firefox.options import Options else: from selenium.webdriver import Chrome as Driver from selenium.webdriver.chrome.options import Options options = Options() options.add_argument("--headless") driver = Driver(options=options) try: driver.get(url) vega_spec = driver.execute_async_script(code, vegalite_spec) finally: driver.close() print("-------------------------------") print(f"Full output for {driver_name}") print(json.dumps(vega_spec, indent=2)) results[driver_name] = vega_spec print("----------------------------") for driver_name, vega_spec in results.items(): print( f'{driver_name}: spec.scales[1].range = {vega_spec["scales"][1]["range"]}' )