def METHOD_NAME(self, scheme_or_url: str) -> Optional[Type[Driver]]:
    """Find the driver class based on scheme or URL
    Args:
        scheme_or_url: The scheme or the url
    Returns:
        The driver instance or None if not found
    """
    index = scheme_or_url.find(":")
    if index > 0:
        scheme = scheme_or_url[0:index]
    else:
        scheme = scheme_or_url
    return self.drivers.get(scheme.lower())