# Pickle data
with open(project_root / "{{ dataset_path }}", "rb") as file:
    data = pickle.load(file)

# Convert to DataFrame if needed
if isinstance(data, pd.DataFrame):
    df = data
else:
    df = pd.DataFrame(data)
