combine const and judge

constants.py

@@ -1,23 +0,0 @@
-# Source: https://github.com/wayveai/LingoQA/blob/main/benchmark/constants.py
-
-"""
-LingoQA datasets are stored in Google Cloud.
-This file provides the download link for the datasets, as well as reference keys for the data.
-"""
-from enum import Enum
-
-LINGOQA_TEST = "https://drive.usercontent.google.com/u/1/uc?id=1I8u6uYysQUstoVYZapyRQkXmOwr-AG3d&export=download"
-
-LINGO_JUDGE = "wayveai/Lingo-Judge"
-
-class Keys(str, Enum):
-    question_id = "question_id"
-    segment_id = "segment_id"
-    question = "question"
-    answer = "answer"
-    references = "references"
-    prediction = "prediction"
-    max_score = "max_score"
-    score = "score"
-    probability = "probability"
-    correct = "correct"

judge.py

@@ -1,11 +1,28 @@
 # Source: https://github.com/wayveai/LingoQA/blob/main/benchmark/judge.py
 
+from enum import Enum
+from typing import List
+
 import torch
 from torch import nn
-
-from typing import List
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
-from .constants import LINGO_JUDGE
+
+LINGOQA_TEST = "https://drive.usercontent.google.com/u/1/uc?id=1I8u6uYysQUstoVYZapyRQkXmOwr-AG3d&export=download"
+
+LINGO_JUDGE = "wayveai/Lingo-Judge"
+
+
+class Keys(str, Enum):
+    question_id = "question_id"
+    segment_id = "segment_id"
+    question = "question"
+    answer = "answer"
+    references = "references"
+    prediction = "prediction"
+    max_score = "max_score"
+    score = "score"
+    probability = "probability"
+    correct = "correct"
 
 
 class LingoJudge(nn.Module):