diff --git a/.github/ci.yml b/.github/workflows/ci.yml similarity index 100% rename from .github/ci.yml rename to .github/workflows/ci.yml diff --git a/.github/publish.yml b/.github/workflows/publish.yml similarity index 100% rename from .github/publish.yml rename to .github/workflows/publish.yml diff --git a/elorus/client.py b/elorus/client.py index ec5ff8d..3289647 100644 --- a/elorus/client.py +++ b/elorus/client.py @@ -49,7 +49,7 @@ def handle_file_download(self, response): filename = filename_kv[1] return filename, response.content - def _handle_response(self, response): + def _handle_response(self, response: httpx.Response): if response.headers.get("Content-Type", "").lower() == "application/pdf": return self.handle_file_download(response) @@ -108,7 +108,7 @@ def _handle_response(self, response): def _handle_request( self, method: str, path: str, payload: Optional[dict] = None, **kwargs - ): + ) -> httpx.Response: auth = self._get_auth() url = f"{self.base_url}/{self.api_version}/{path}" with httpx.Client(auth=auth) as client: @@ -116,8 +116,8 @@ def _handle_request( return self._handle_response(response) -class SubClient: - client = Client +class SubClient(Client): + client: Client def __init__(self, client: Client): self.client = client