Fix lattice
Created by: VolodyaCO
Couldn't reopen #613. I just put the tolerance of kdtree cutoff inside the get_edges method. And also provided a bit more custom arguments for drawing the graph, while making sure that lattices are 1d or 2d in order to be drawn.