Spaces:
Running
on
T4
Running
on
T4
Ahmet Kaan Sever
commited on
Commit
·
26c03b8
1
Parent(s):
8e04a46
forced dataset sizes to be 3 for testing
Browse files
src/deepeval/base_task.py
CHANGED
@@ -188,7 +188,7 @@ class BaseTask(ABC):
|
|
188 |
print("Loading dataset from Hugging Face.")
|
189 |
dataset= load_dataset(self.dataset_repo, token=HF_TOKEN, split="train")
|
190 |
print("Dataset loaded.")
|
191 |
-
return dataset
|
192 |
|
193 |
@abstractmethod
|
194 |
def evaluate(self):
|
|
|
188 |
print("Loading dataset from Hugging Face.")
|
189 |
dataset= load_dataset(self.dataset_repo, token=HF_TOKEN, split="train")
|
190 |
print("Dataset loaded.")
|
191 |
+
return dataset.select(range(min(3, len(dataset))))
|
192 |
|
193 |
@abstractmethod
|
194 |
def evaluate(self):
|