from qlbm.components.common import UniformStatePrep

UniformStatePrep(4, 5).draw("mpl")