[Kwant] How to set the grid in kwant.continuum.discretize?