From 020c79970f49a599402befc475c2bd18c275146b Mon Sep 17 00:00:00 2001 From: Vitor Avila <96086495+Vitor-Avila@users.noreply.github.com> Date: Wed, 29 May 2024 16:05:48 -0300 Subject: [PATCH] fix(reports): Update the element class to wait for when taking a screenshot (#28745) --- superset/utils/webdriver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/superset/utils/webdriver.py b/superset/utils/webdriver.py index fc7f18aca..d79a9c746 100644 --- a/superset/utils/webdriver.py +++ b/superset/utils/webdriver.py @@ -187,7 +187,7 @@ class WebDriverPlaywright(WebDriverProxy): try: # chart containers didn't render logger.debug("Wait for chart containers to draw at url: %s", url) - slice_container_locator = page.locator(".slice_container") + slice_container_locator = page.locator(".chart-container") slice_container_locator.first.wait_for() for slice_container_elem in slice_container_locator.all(): slice_container_elem.wait_for() @@ -375,7 +375,7 @@ class WebDriverSelenium(WebDriverProxy): logger.debug("Wait for chart containers to draw at url: %s", url) WebDriverWait(driver, self._screenshot_locate_wait).until( EC.visibility_of_all_elements_located( - (By.CLASS_NAME, "slice_container") + (By.CLASS_NAME, "chart-container") ) ) except TimeoutException as ex: