ott.geometry.geometry.Geometry.set_scale_cost