Fixing defaultdict -> collections.defaultdict

AI-Hypercomputer · Jan 13, 2025 · 53ec9a3 · 53ec9a3
1 parent c1e9124
commit 53ec9a3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/MaxText/benchmarks/mmlu/mmlu_eval.py b/MaxText/benchmarks/mmlu/mmlu_eval.py
@@ -176,8 +176,8 @@ def main(config):
     max_logging.log(f"Accuracy for subcategory '{subcat_label}': {acc:.4f}")
 
   # Calculate and print category accuracies
-  cat_correct = defaultdict(int)
-  cat_total = defaultdict(int)
+  cat_correct = collections.defaultdict(int)
+  cat_total = collections.defaultdict(int)
 
   for category_name, subcat_labels in categories.items():
     for subcat_label in subcat_labels: