* Adds pretrained IDs directly in the tests * Fix tests * Fix tests * Review!
get_values(MODEL_MAPPING)