fix test

Author: shivaji-kharse

tok/hnsw/persistent_hnsw.go

@@ -354,7 +354,6 @@ func (ph *persistentHNSW[T]) MergeResults(ctx context.Context, c index.CacheType
 	})
 
 	uids := []uint64{}
-	// out of range error
 	for i := range maxResults {
 		if i > len(result) {
 			break

tok/partitioned_hnsw/partitioned_hnsw.go

@@ -209,6 +209,11 @@ func (ph *partitionedHNSW[T]) Search(ctx context.Context, txn index.CacheType, q
 		}(index)
 	}
 	wg.Wait()
+
+	if len(res) == 0 {
+		return res, nil
+	}
+
 	return ph.clusterMap[0].MergeResults(ctx, txn, res, query, maxResults, filter)
 }
 

worker/task.go

@@ -368,7 +368,7 @@ func (qs *queryState) handleValuePostings(ctx context.Context, args funcArgs) er
 
 		nnUids, err := indexer.Search(ctx, qc, srcFn.vectorInfo,
 			int(numNeighbors), index.AcceptAll[float32])
-		if err != nil {
+		if err != nil && !strings.Contains(err.Error(), hnsw.EmptyHNSWTreeError+": "+badger.ErrKeyNotFound.Error()) {
 			return err
 		}