class HfHubHTTPError(HTTPError, OSError):

    def __init__(
        self,
        message: str,
        *,
        response: Response,
        server_message: Optional[str] = None,
    ):
        self.request_id = response.headers.get("x-request-id") or response.headers.get("X-Amzn-Trace-Id")
        self.server_message = server_message
        self.response = response
        self.request = response.request
        super().__init__(message)

    def __reduce_ex__(self, protocol):
        """Fix pickling of Exception subclass with kwargs. We need to override __reduce_ex__ of the parent class"""
        return (self.__class__, (str(self),), {"response": self.response, "server_message": self.server_message})

# test_hf_cloudpickle_bug.py
from copy import deepcopy
from huggingface_hub import HfFileSystem
from huggingface_hub.utils import RepositoryNotFoundError
from requests import Response, Request

# Mock an error
resp = Response()
resp.status_code = 404
resp.url = "https://huggingface.co/api/datasets/rotten_tomatoes/test.parquet"
resp.request = Request("GET", "https://huggingface.co/api/datasets/rotten_tomatoes/test.parquet")
resp._content = b'{"error": "Repository Not Found"}'

err = RepositoryNotFoundError(
    "404 Client Error. Repository Not Found.",
    response=resp,
    server_message="Repository Not Found",
)

fs = HfFileSystem()
# Simulate the error in cache
fs._repo_and_revision_exists_cache = {
    ("dataset", "rotten_tomatoes/test.parquet", None): (False, err),
}

# Now try to deepcopy the cache: this is exactly what _get_instance_state does.
cache_copy = deepcopy(fs._repo_and_revision_exists_cache)  # <- expected to fail on buggy behavior