Changed 100 to 50

src/deepeval/base_task.py

@@ -204,7 +204,7 @@ class BaseTask(ABC):
         )
 
         inputs = self.tokenizer(formatted_chat, return_tensors="pt", padding=True, truncation=True)
-        
+
         if self.device == "auto":
             input_ids = inputs.input_ids
             attention_mask = inputs.attention_mask
@@ -246,8 +246,8 @@ class BaseTask(ABC):
         print("Dataset loaded.")
     
         # Load 50 from each dataset
-        if len(dataset) > 100:
-            dataset = dataset.shuffle(seed=42).select(range(100))
+        if len(dataset) > 50:
+            dataset = dataset.shuffle(seed=42).select(range(50))
         end_time = datetime.now()
         print(f"Dataset loaded in {(end_time - start_time).seconds} seconds.")
         return dataset