fixing typing for namespaced aio import

shabbyrobe · Oct 18, 2023 · 123e7e2 · 123e7e2
1 parent 94f7aee
commit 123e7e2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/grpc-stubs/__init__.pyi b/grpc-stubs/__init__.pyi
@@ -4,6 +4,8 @@ import typing
 from concurrent import futures
 from types import ModuleType, TracebackType
 
+from grpc import aio
+
 __version__: str
 
 # This class encodes an uninhabited type, requiring use of explicit casts or ignores

diff --git a/typesafety/test_grpc.yml b/typesafety/test_grpc.yml
@@ -12,6 +12,13 @@
     reveal_type(Channel())  # N: Revealed type is "grpc.aio.Channel"
     reveal_type(Server())   # N: Revealed type is "grpc.aio.Server"
 
+- case: grpc_aio_package
+  main: |
+    import grpc
+
+    reveal_type(grpc.aio.Channel())  # N: Revealed type is "grpc.aio.Channel"
+    reveal_type(grpc.aio.Server())   # N: Revealed type is "grpc.aio.Server"
+
 - case: grpc_status
   main: |
     from grpc import Status