From c96fbe3afc521b3772408b0727b8d8ae36e01d2c Mon Sep 17 00:00:00 2001
From: Fahim Imaduddin Dalvi <faimaduddin@hbku.edu.qa>
Date: Mon, 11 Sep 2023 14:05:07 +0300
Subject: [PATCH] Fix incorrect param handling in dataset_base init

---
 llmebench/datasets/dataset_base.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/llmebench/datasets/dataset_base.py b/llmebench/datasets/dataset_base.py
index adc92b19..a95ff52e 100644
--- a/llmebench/datasets/dataset_base.py
+++ b/llmebench/datasets/dataset_base.py
@@ -24,7 +24,9 @@ class DatasetBase(ABC):
 
     Attributes
     ----------
-        None
+    data_dir : str
+        Base path of data containing all datasets. Defaults to "data" in the current
+        working directory.
 
     Methods
     -------
@@ -50,8 +52,8 @@ class DatasetBase(ABC):
 
     """
 
-    def __init__(self, **kwargs):
-        self.data_dir = kwargs.get("data_dir", "data")
+    def __init__(self, data_dir="data", **kwargs):
+        self.data_dir = data_dir
 
     @abstractmethod
     def metadata(self):