Add abstract constructor

988a400f · Leo Gao · b0585de4 · 988a400f
Commit 988a400f authored Oct 05, 2020 by Leo Gao
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

lm_eval/base.py lm_eval/base.py +3 -0

No files found.
--- a/lm_eval/base.py
+++ b/lm_eval/base.py
@@ -54,6 +54,9 @@ class LM(abc.ABC):


 class Dataset(abc.ABC):
+    @abc.abstractmethod
+    def __init__(self):
+        self.download()

    def download(self):
        """Downloads the task dataset if necessary"""