fix: cleanup

jina/drivers/search.py

@@ -121,9 +121,6 @@ def _apply_all(self, docs: 'DocumentSet', *args, **kwargs) -> None:
             self.logger.warning(f'these bad docs can not be added: {bad_docs}')
         idx, dist = self.exec_fn(embed_vecs, top_k=int(self.top_k))
 
-        if idx is None and dist is None:
-            return
-
         op_name = self.exec.__class__.__name__
         # can be None if index is size 0
         if idx is not None and dist is not None:

tests/integration/docidcache/test_crud_cache.py

@@ -16,11 +16,9 @@
 VEC_IDX_FILENAME = 'vec_idx.bin'
 DOCS_TO_SEARCH = 1
 TOP_K = 5
-BATCH_SIZE = 4
+REQUEST_SIZE = 4
 DOCS_TO_INDEX = 10
 
-pytest.register_assert_rewrite('check_indexers_size')
-
 
 def config_env(field, tmp_workspace, shards, indexers, polling):
     os.environ['JINA_SHARDS'] = str(shards)
@@ -71,11 +69,6 @@ def get_documents(chunks, same_content, nr=10, index_start=0):
         yield d
 
 
-docs_chunks = [0, 3, 5]
-docs_same_content = [False, True]
-docs_nr = [0, 10, 100]
-
-
 def get_index_flow(field, tmp_path, shards, indexers):
     config_env(field, tmp_path, shards, indexers, polling='any')
     f = Flow.load_config(os.path.join(os.curdir, 'crud_cache_flow_index.yml'))
@@ -245,7 +238,7 @@ def validate_results(resp):
 
     # INDEX
     with flow_index as f:
-        f.index(docs, batch_size=BATCH_SIZE)
+        f.index(docs, request_size=REQUEST_SIZE)
 
     check_indexers_size(chunks, len(docs), field, tmp_path, same_content, shards, 'index')
 
@@ -255,7 +248,7 @@ def validate_results(resp):
 
     new_docs = list(get_documents(chunks=chunks, same_content=same_content, index_start=index_start_new_docs))
     with flow_index as f:
-        f.index(new_docs, batch_size=BATCH_SIZE)
+        f.index(new_docs, request_size=REQUEST_SIZE)
 
     check_indexers_size(chunks, len(docs), field, tmp_path, same_content, shards, 'index2')