Re:Using Python and Selenium for automated visual regression...
by seleniumbase on Wednesday February 12, 2020 @09:36PM
(#22314858)
SeleniumBase has those retries built-in. The browser will automatically wait for page elements to fully load before interacting with them. In addition, SeleniumBase includes all the abilities of pytest, which transforms Selenium from a library into a complete test automation framework.
I don't know what specific problems you were having, so I can't give more specific advice than that.