When pyplot.imshow() is used without extent option, the pixels are shown on integer coordinates. All pixels are fit into the defined range. As a result, the plot is a bit smaller than it should be.

If function returns matplotlib.pyplot.Figure object, it cannot be cached with st.cache. While caching the return value of mk_figure(), Streamlit encountered an object of type matplotlib.spines.Spines, ...