[Kwant] Re: Figsize when ax is enabled in kwant.plot