refactor: Fix type hints and variable references in gpytranslate

DavideGalilei · Dec 3, 2024 · ea857df · ea857df
1 parent 7375acd
commit ea857df
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/gpytranslate/gpytranslate.py b/gpytranslate/gpytranslate.py
@@ -149,7 +149,7 @@ async def translate(
                 client: httpx.AsyncClient
                 raw: Union[Mapping, List] = (
                     (
-                        await c.post(
+                        await client.post(
                             self.url,
                             params={**params, "q": text},
                             headers=self.get_headers(),
@@ -159,7 +159,7 @@ async def translate(
                     else (
                         {
                             k: (
-                                await c.post(
+                                await client.post(
                                     self.url,
                                     params={**params, "q": v},
                                     headers=self.get_headers(),
@@ -170,7 +170,7 @@ async def translate(
                         if isinstance(text, Mapping)
                         else [
                             (
-                                await c.post(
+                                await client.post(
                                     self.url,
                                     params={**params, "q": elem},
                                     headers=self.get_headers(),
@@ -180,7 +180,7 @@ async def translate(
                         ]
                     )
                 )
-                await c.aclose()
+                await client.aclose()
 
             return self.check(raw=raw, client=client, dt=dt, text=text)
         except Exception as e:
@@ -238,11 +238,11 @@ async def tts(
                 ) as resp:
                     resp: httpx.Response
                     if isinstance(file, io.BytesIO):
-                        async for chunk in response.aiter_bytes(chunk_size=chunk_size):
+                        async for chunk in resp.aiter_bytes(chunk_size=chunk_size):
                             file.write(chunk)
                     else:
                         file: AsyncBufferedIOBase
-                        async for chunk in response.aiter_bytes(chunk_size=chunk_size):
+                        async for chunk in resp.aiter_bytes(chunk_size=chunk_size):
                             await file.write(chunk)
                 await c.aclose()
 

diff --git a/gpytranslate/types/base_translator.py b/gpytranslate/types/base_translator.py
@@ -1,5 +1,5 @@
 from collections.abc import Mapping
-from typing import Any, Callable, Dict, List, Union
+from typing import Any, Callable, Dict, List, Optional, Union
 
 from .translated_object import TranslatedObject