coh mech

ccl_win.py

@@ -115,7 +115,7 @@ class ccl_win(evaluate.Measurement):
 
 
 
-    def _compute(self, predictions, dataset, batch_size: int = 16, device=None, use_aggregator=True):
+    def _compute(self, predictions, dataset="arxiv", batch_size: int = 16, device=None, use_aggregator=True):
         """Returns the scores"""
         MODEL_CACHE_DIR = "/home/rcardena/.cache/huggingface/"
         BASEDIR = "/gfs/team/nlp/users/rcardena/tools/new_evals/ccl_win"