def _driver_class(self): if os.environ.get("DISABLE_XVFB_DRIVER"): return Port._driver_class(self) return XvfbDriver