augment_data_test.py

from utils.data_utils import augment_data_with_llm, add_column_metadata
import json


# with open('data/id_to_metadata.json', 'r') as f:
#     id_to_metadata = json.load(f)
# add_column_metadata(id_to_metadata)
with open('data/id_to_metadata_col.json', 'r') as f:
    id_to_metadata_col = json.load(f)
augment_data_with_llm(id_to_metadata_col)