this has potential, pls update this ! !
(THere's a bugfixed version on dimod, i'll port it here too once i can)
