Skip to content

Commit ce2ee8d

Browse files
authored
try to catch failing webdrivers
1 parent 8323f80 commit ce2ee8d

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

pypop/notebook_interface/plotting.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -109,10 +109,13 @@ def _repr_png_(self):
109109
self.figure.toolbar_location = None
110110

111111
driver.set_window_size(*window_size)
112-
113-
img = get_screenshot_as_png(self.figure, driver=driver,)
114-
115-
driver.quit()
112+
113+
try:
114+
img = get_screenshot_as_png(self.figure, driver=driver, timeout=10)
115+
except:
116+
driver.quit()
117+
warn("Webdriver failed - falling back to matplotlib interface")
118+
return self._mpl_fallback_png()
116119

117120
imgbuffer = BytesIO()
118121
img.save(imgbuffer, format="png")

0 commit comments

Comments
 (0)