Add Gar-T5 topics to prebuilt (#2044)

src/main/java/io/anserini/search/topicreader/Topics.java

@@ -121,6 +121,14 @@ public enum Topics {
   DPR_SQUAD_TEST(DprJsonlTopicReader.class, "topics-and-qrels/topics.dpr.squad.test.txt"),
   NQ_DEV(DprNqTopicReader.class, "topics-and-qrels/topics.nq.dev.txt"),
   NQ_TEST(DprNqTopicReader.class, "topics-and-qrels/topics.nq.test.txt"),
+  NQ_TEST_GART5_ANSWERS(TsvIntTopicReader.class, "topics-and-qrels/topics.nq.test.gar-t5.answers.tsv"),
+  NQ_TEST_GART5_TITLES(TsvIntTopicReader.class, "topics-and-qrels/topics.nq.test.gar-t5.titles.tsv"),
+  NQ_TEST_GART5_SENTENCES(TsvIntTopicReader.class, "topics-and-qrels/topics.nq.test.gar-t5.sentences.tsv"),
+  NQ_TEST_GART5_ALL(TsvIntTopicReader.class, "topics-and-qrels/topics.nq.test.gar-t5.all.tsv"),
+  DPR_TRIVIA_TEST_GART5_ANSWERS(TsvIntTopicReader.class, "topics-and-qrels/topics.dpr.trivia.test.gar-t5.answers.tsv"),
+  DPR_TRIVIA_TEST_GART5_TITLES(TsvIntTopicReader.class, "topics-and-qrels/topics.dpr.trivia.test.gar-t5.titles.tsv"),
+  DPR_TRIVIA_TEST_GART5_SENTENCES(TsvIntTopicReader.class, "topics-and-qrels/topics.dpr.trivia.test.gar-t5.sentences.tsv"),
+  DPR_TRIVIA_TEST_GART5_ALL(TsvIntTopicReader.class, "topics-and-qrels/topics.dpr.trivia.test.gar-t5.all.tsv"),
   MRTYDI_V11_AR_TRAIN(TsvIntTopicReader.class, "topics-and-qrels/topics.mrtydi-v1.1-ar.train.txt.gz"),
   MRTYDI_V11_AR_DEV(TsvIntTopicReader.class, "topics-and-qrels/topics.mrtydi-v1.1-ar.dev.txt.gz"),
   MRTYDI_V11_AR_TEST(TsvIntTopicReader.class, "topics-and-qrels/topics.mrtydi-v1.1-ar.test.txt.gz"),

src/test/java/io/anserini/search/topicreader/TopicReaderTest.java

@@ -37,7 +37,7 @@ public void testIterateThroughAllEnums() {
       String[] pathParts = topic.path.split("/");
       assertEquals(topic.readerClass, TopicReader.getTopicReaderClassByFile(pathParts[1]));
     }
-    assertEquals(307, cnt);
+    assertEquals(315, cnt);
   }
 
   @Test
@@ -617,6 +617,22 @@ public void testNq() {
     assertEquals("who sings gimme shelter with mick jagger", topics.get(1756).get("title"));
   }
 
+  @Test
+  public void testGarT5Nq() {
+    assertEquals(3610, TopicReader.getTopics(Topics.NQ_TEST_GART5_ANSWERS).keySet().size());
+    assertEquals(3610, TopicReader.getTopics(Topics.NQ_TEST_GART5_TITLES).keySet().size());
+    assertEquals(3610, TopicReader.getTopics(Topics.NQ_TEST_GART5_SENTENCES).keySet().size());
+    assertEquals(3610, TopicReader.getTopics(Topics.NQ_TEST_GART5_ALL).keySet().size());
+  }
+
+  @Test
+  public void testGarT5Trivia() {
+    assertEquals(11313, TopicReader.getTopics(Topics.DPR_TRIVIA_TEST_GART5_ANSWERS).keySet().size());
+    assertEquals(11313, TopicReader.getTopics(Topics.DPR_TRIVIA_TEST_GART5_TITLES).keySet().size());
+    assertEquals(11313, TopicReader.getTopics(Topics.DPR_TRIVIA_TEST_GART5_SENTENCES).keySet().size());
+    assertEquals(11313, TopicReader.getTopics(Topics.DPR_TRIVIA_TEST_GART5_ALL).keySet().size());
+  }
+
   @Test
   public void testTREC19DL() {
     SortedMap<Integer, Map<String, String>> topics;