Incorporate training prior into config

[EthanMarx]

Currently the exact training prior isn't saved into wandb or the config. Only the function name that defines the prior is. This can lead to discrepancies if the training prior is re-defined w/o changing the function name used. We should incorporate building the training prior in the config such that it can be saved on wandb and in the local configs.