Ahmet Kaan Sever commited on
Commit
26c03b8
·
1 Parent(s): 8e04a46

forced dataset sizes to be 3 for testing

Browse files
Files changed (1) hide show
  1. src/deepeval/base_task.py +1 -1
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):