test.py

import pandas as pd

# Assuming your CSV file is named 'data.csv'
try:
    df = pd.read_csv('data/bbc_news_4o_mini.csv')
    # df = pd.read_csv('data/MAGE_4o_mini.csv')
    
    print(df.columns)  # header names
    print(len(df))

except FileNotFoundError:
    print("Error: data.csv not found")
except Exception as e:
    print(f"An error occurred: {e}")