Skip to content

Commit

Permalink
Remove blobfile usage from MMLU
Browse files Browse the repository at this point in the history
Inspired by: openai#4
  • Loading branch information
lucasresck committed Nov 11, 2024
1 parent 580d359 commit e0bab8a
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions mmlu_eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@
import random
import re

import blobfile as bf
import pandas

from . import common
Expand Down Expand Up @@ -88,7 +87,7 @@ def __init__(self, num_examples: int | None = None, language: str = "EN-US"):
url = f"https://openaipublic.blob.core.windows.net/simple-evals/mmlu_{language}.csv"
else:
url = "https://openaipublic.blob.core.windows.net/simple-evals/mmlu.csv"
df = pandas.read_csv(bf.BlobFile(url))
df = pandas.read_csv(url)
examples = [row.to_dict() for _, row in df.iterrows()]
if num_examples:
examples = random.Random(0).sample(examples, num_examples)
Expand Down

0 comments on commit e0bab8a

Please sign in to comment.