Fix type anotation

saghul · Oct 6, 2023 · a443dcf · a443dcf
1 parent 7c321d8
commit a443dcf
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/aiodns/__init__.py b/aiodns/__init__.py
@@ -9,7 +9,7 @@
     List,
     Optional,
     Set,
-    Union
+    Sequence
 )
 
 from . import error
@@ -45,7 +45,7 @@
                    }
 
 class DNSResolver:
-    def __init__(self, nameservers: Optional[List[str]] = None,
+    def __init__(self, nameservers: Optional[Sequence[str]] = None,
                  loop: Optional[asyncio.AbstractEventLoop] = None,
                  **kwargs: Any) -> None:
         self.loop = loop or asyncio.get_event_loop()
@@ -59,7 +59,7 @@ def __init__(self, nameservers: Optional[List[str]] = None,
         self._timer = None  # type: Optional[asyncio.TimerHandle]
 
     @property
-    def nameservers(self) -> List[Union[str, bytes]]:
+    def nameservers(self) -> List[str]:
         return self._channel.servers
 
     @nameservers.setter
@@ -75,7 +75,7 @@ def _callback(fut: asyncio.Future, result: Any, errorno: int) -> None:
         else:
             fut.set_result(result)
 
-    def query(self, host: str, qtype: str, qclass: str=None) -> asyncio.Future:
+    def query(self, host: str, qtype: str, qclass: Optional[str]=None) -> asyncio.Future:
         try:
             qtype = query_type_map[qtype]
         except KeyError: