wait for element grrs002

Author: T4rk1n

components/dash-core-components/tests/integration/graph/test_graph_responsive.py

@@ -205,7 +205,7 @@ def update_figures(n_clicks):
     dash_dcc.wait_for_text_to_equal("#generate-btn", "Generate Figures")
     dash_dcc.find_element("#generate-btn").click()
     dash_dcc.wait_for_text_to_equal("#bounding-output", "loaded")
-    dash_dcc.find_element(".dash-graph .js-plotly-plot.dash-graph--pending")
-    dash_dcc.find_element(".dash-graph .js-plotly-plot:not(.dash-graph--pending)")
+    dash_dcc.wait_for_element(".dash-graph .js-plotly-plot.dash-graph--pending")
+    dash_dcc.wait_for_element(".dash-graph .js-plotly-plot:not(.dash-graph--pending)")
     dash_dcc.find_element("#bounding-btn").click()
     dash_dcc.wait_for_text_to_equal("#bounding-output", "true")