from vireoSNP.plot import heat_matrix
import numpy as np
np.random.seed(1)
X = np.random.rand(5, 7)
heat_matrix(X)