Improve doc for pde_aware_translation_cost_model