Fix partially unknown typing errors (#181)

jborean93 · Nov 9, 2023 · d2a3eca · d2a3eca
1 parent 17fe1e2
commit d2a3eca
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/psrp/_async.py b/src/psrp/_async.py
@@ -1041,7 +1041,7 @@ async def connect_async(
 
     async def invoke(
         self,
-        input_data: t.Optional[t.Union[t.Iterable, t.AsyncIterable]] = None,
+        input_data: t.Optional[t.Union[t.Iterable[t.Any], t.AsyncIterable[t.Any]]] = None,
         output_stream: t.Optional[AsyncPSDataCollection[t.Any]] = None,
         buffer_input: bool = True,
     ) -> t.List[t.Any]:
@@ -1093,7 +1093,7 @@ async def invoke(
 
     async def invoke_async(
         self,
-        input_data: t.Optional[t.Union[t.Iterable, t.AsyncIterable]] = None,
+        input_data: t.Optional[t.Union[t.Iterable[t.Any], t.AsyncIterable[t.Any]]] = None,
         output_stream: t.Optional[AsyncPSDataCollection[t.Any]] = None,
         completed: t.Optional[t.Callable[[], t.Awaitable[None]]] = None,
         buffer_input: bool = True,
@@ -1158,8 +1158,8 @@ async def invoke_async(
             log.debug("Sending pipeline input - %s", self._pipeline.pipeline_id)
 
             async def input_gen(
-                input_data: t.Union[t.Iterable, t.AsyncIterable],
-            ) -> t.AsyncIterator:
+                input_data: t.Union[t.Iterable[t.Any], t.AsyncIterable[t.Any]],
+            ) -> t.AsyncIterator[t.Any]:
                 if isinstance(input_data, t.Iterable):
                     for data in input_data:
                         yield data
@@ -1581,7 +1581,7 @@ def add_statement(self) -> "AsyncPowerShell":
 
     async def invoke_async(
         self,
-        input_data: t.Optional[t.Union[t.Iterable, t.AsyncIterable]] = None,
+        input_data: t.Optional[t.Union[t.Iterable[t.Any], t.AsyncIterable[t.Any]]] = None,
         output_stream: t.Optional[AsyncPSDataCollection[t.Any]] = None,
         completed: t.Optional[t.Callable[[], t.Awaitable[None]]] = None,
         buffer_input: bool = True,

diff --git a/src/psrp/_sync.py b/src/psrp/_sync.py
@@ -1091,7 +1091,7 @@ def connect_async(
 
     def invoke(
         self,
-        input_data: t.Optional[t.Iterable] = None,
+        input_data: t.Optional[t.Iterable[t.Any]] = None,
         output_stream: t.Optional[SyncPSDataCollection[t.Any]] = None,
         buffer_input: bool = True,
     ) -> t.List[t.Any]:
@@ -1142,7 +1142,7 @@ def invoke(
 
     def invoke_async(
         self,
-        input_data: t.Optional[t.Iterable] = None,
+        input_data: t.Optional[t.Iterable[t.Any]] = None,
         output_stream: t.Optional[SyncPSDataCollection[t.Any]] = None,
         completed: t.Optional[t.Callable[[], None]] = None,
         buffer_input: bool = True,