removed unused classes

OpenGPTX · Feb 20, 2024 · 35fb949 · 35fb949
1 parent f7380e7
commit 35fb949
Showing 1 changed file with 1 addition and 13 deletions.
diff --git a/lm_eval/tasks/opengptx/arcx.py b/lm_eval/tasks/opengptx/arcx.py
@@ -134,16 +134,4 @@ def should_decontaminate(self):
         return True
 
     def doc_to_decontamination_query(self, doc):
-        return doc["query"]
-
-
-class ARCChallenge(ARCBase):
-    def __init__(self, lang: str, **kwargs):
-        self.DATASET_NAME = f"challenge_{lang.upper()}"
-        super().__init__(**kwargs)
-
-
-class ARCEasy(ARCBase):
-    def __init__(self, lang: str, **kwargs):
-        self.DATASET_NAME = f"easy_{lang.upper()}"
-        super().__init__(**kwargs)
+        return doc["query"]