util.py

import pandas as pd

def read_data(file_path):
    """
        Read data from a csv file
    """
    return pd.read_csv(file_path, sep=",")


if __name__ == "__main__":
    file_path = "data/mmlu/response_rec.csv"
    data = read_data(file_path)
    high_school_physics = data[data['sub_topic'] == 'high_school_physics'] 
    print(high_school_physics.head(5))