fix: typo

jina-ai · Sep 16, 2022 · f4ee3a2 · f4ee3a2
1 parent 2697488
commit f4ee3a2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/client/clip_client/client.py b/client/clip_client/client.py
@@ -492,7 +492,9 @@ def rank(
 
         return results
 
-    async def arank(self, docs: Iterable['Document'], **kwargs) -> 'DocumentArray':
+    async def arank(
+        self, docs: Union['DocumentArray', Iterable['Document']], **kwargs
+    ) -> 'DocumentArray':
         from rich import filesize
 
         if hasattr(docs, '__len__') and len(docs) == 0:

diff --git a/tests/test_ranker.py b/tests/test_ranker.py
@@ -213,7 +213,7 @@ def test_docarray_inputs(make_flow, inputs):
 @pytest.mark.asyncio
 async def test_async_arank(make_flow, inputs):
     c = Client(server=f'grpc://0.0.0.0:{make_flow.port}')
-    r = await c.arank(inputs)
+    r = await c.arank(inputs if not callable(inputs) else inputs())
     assert '__loaded_by_CAS__' not in r[0].tags
     assert not r[0].blob
     assert not r[0].tensor