From e40c22adce1c7f4d7300ba81e2c62b2950c9b100 Mon Sep 17 00:00:00 2001
From: Andrew Parry <a.parry.1@research.gla.ac.uk>
Date: Mon, 11 Nov 2024 09:26:38 +0000
Subject: [PATCH] should be false, listwise is an edge case

---
 rankers/datasets/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/rankers/datasets/dataset.py b/rankers/datasets/dataset.py
index d7aaabc..d9f524b 100644
--- a/rankers/datasets/dataset.py
+++ b/rankers/datasets/dataset.py
@@ -16,7 +16,7 @@ def __init__(self,
                  group_size: int = 2,
                  no_positive: bool = False,
                  lazy_load_text : bool = True,
-                 top_k_group : bool = True
+                 top_k_group : bool = False,
                  ) -> None:
         assert training_dataset_file.endswith('jsonl'), "Training dataset should be a JSONL file and should not be compressed"