src/utils/scale.py

import pickle

def scale_memory(context, query, scale=3):
    with open(query, "rb") as f:
        query_metas = pickle.load(f)

    labels_contents = [ref for q_meta  in query_metas for ref in q_meta['reference_list']]
    copy_reference = [f"{ref}_copy{bias}" for bias, ref in enumerate(context * scale) if ref not in labels_contents]
    context.extend(copy_reference)