Addressing Martin Comments

Signed-off-by: Varun Jain <[email protected]>
opensearch-project · Jul 9, 2024 · ec029af · ec029af
1 parent 8c7ed1a
commit ec029af
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/src/main/java/org/opensearch/neuralsearch/search/query/HybridCollectorManager.java b/src/main/java/org/opensearch/neuralsearch/search/query/HybridCollectorManager.java
@@ -171,18 +171,15 @@ private TopDocsAndMaxScore getTopDocsAndAndMaxScore(
         final DocValueFormat[] docValueFormats
     ) {
         TopDocs newTopDocs;
-        List<? extends TopDocs> topDocs = hybridSearchCollector.topDocs();
+        List topDocs = hybridSearchCollector.topDocs();
         if (docValueFormats != null) {
             newTopDocs = getNewTopFieldDocs(
                 getTotalHits(this.trackTotalHitsUpTo, topDocs, hybridSearchCollector.getTotalHits()),
-                (List<TopFieldDocs>) topDocs,
+                topDocs,
                 sortAndFormats.sort.getSort()
             );
         } else {
-            newTopDocs = getNewTopDocs(
-                getTotalHits(this.trackTotalHitsUpTo, topDocs, hybridSearchCollector.getTotalHits()),
-                (List<TopDocs>) topDocs
-            );
+            newTopDocs = getNewTopDocs(getTotalHits(this.trackTotalHitsUpTo, topDocs, hybridSearchCollector.getTotalHits()), topDocs);
         }
         return new TopDocsAndMaxScore(newTopDocs, hybridSearchCollector.getMaxScore());
     }