Remove 'model' key when parsing interaction_graph_info
Description
Fixes the incorrect key in the model_config_dict in run_mlp_ablations.py. Assumes both lms and mlps store the pure model config (and no other info such as train info) in their results. See discussion in #259.
Creates a new sample graph (which will store just the model config inside model_config_dict. This should have been done in #235 when this breaking change was introduced.
Remove 'model' key when parsing interaction_graph_info
Description
Related Issue
Closes #259
Does this PR introduce a breaking change?
Yes. Well, it fixes the earlier breaking change.