Update dataset_previews.py

dataset_previews.py

@@ -4,7 +4,7 @@ import pandas as pd
 import numpy as np
 from typing import Dict, Any, List, Tuple
 import collections
-from mmlu_pro_eval_adapted import load_mmlu_pro_data, preprocess  # Import preprocess also
+from mmlu_pro_eval_adapted import load_mmlu_pro, preprocess  # Import preprocess also
 
 def calculate_dataset_statistics():
     """
@@ -15,7 +15,7 @@ def calculate_dataset_statistics():
     """
     try:
         # Load MMLU-Pro data using the function from mmlu_pro_eval_adapted
-        mmlu_data = load_mmlu_pro_data(num_subjects=-1, num_questions=-1)
+        mmlu_data = load_mmlu_pro(num_subjects=-1, num_questions=-1)
         
         # Calculate total questions and questions per subject
         total_questions = 0