fix(cli): fix missing type_checking if statement (#4234)

jina-ai · Feb 4, 2022 · d1a0833 · d1a0833
1 parent 13edc16
commit d1a0833
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/cli/api.py b/cli/api.py
@@ -1,6 +1,7 @@
 from jina.serve.runtimes.head import HeadRuntime
+from typing import TYPE_CHECKING
 
-if False:
+if TYPE_CHECKING:
     from argparse import Namespace
 
 

diff --git a/tests/unit/serve/executors/test_executor.py b/tests/unit/serve/executors/test_executor.py
@@ -272,7 +272,7 @@ async def foo(self, docs: DocumentArray, **kwargs):
     N = 2
     da = DocumentArray.empty(N)
     exec = AsyncExecutor()
-    da1 = await exec.foo(docs=da)
+    da1 = await exec.foo(da)
     assert da1.texts == ['hello'] * N
 
 
@@ -292,5 +292,5 @@ async def foo(self, docs: DocumentArray, **kwargs):
     N = 2
     da = DocumentArray.empty(N)
     exec = AsyncExecutor()
-    da1 = await exec.foo(docs=da)
+    da1 = await exec.foo(da)
     assert da1.texts == ['hello'] * N