refactor: Properly skip searchers when parallel=100

Author: KShivendu

engine/base_client/search.py

@@ -71,21 +71,6 @@ def search_all(
 
         search_one = functools.partial(self.__class__._search_one, top=top)
 
-        if parallel == 100:
-            return {
-                "total_time": 0,
-                "mean_time": 0,
-                "mean_precisions": 0,
-                "std_time": 0,
-                "min_time": 0,
-                "max_time": 0,
-                "rps": 0,
-                "p95_time": 0,
-                "p99_time": 0,
-                "precisions": [],
-                "latencies": [],
-            }
-
         if parallel == 1:
             start = time.perf_counter()
             precisions, latencies = list(

engine/base_client/upload.py

@@ -38,14 +38,6 @@ def upload(
             self.host, distance, self.connection_params, self.upload_params
         )
 
-        if parallel == 100:
-            return {
-                "post_upload": {},
-                "upload_time": 0,
-                "total_time": 0,
-                "latencies": []
-            }
-
         if parallel == 1:
             for batch in iter_batches(tqdm.tqdm(records), batch_size):
                 latencies.append(self._upload_batch(batch))

engine/clients/client_factory.py

@@ -88,6 +88,7 @@ def _create_searchers(self, experiment) -> List[BaseSearcher]:
                 search_params=search_params,
             )
             for search_params in experiment.get("search_params", [{}])
+            if not search_params.get("parallel") == 100
         ]
 
         return engine_searchers