Allow passing max_expansion_radius to with_refinement