Fix up 2D lattice example.

examples/visualization/lattice_2d.py

@@ -1,4 +1,4 @@
-from abp.fancy import GraphState
+from abp import GraphState, VizClient
 from abp.util import xyz
 import numpy as np
 import time
@@ -33,12 +33,13 @@ def lattice(unit_cell, size):
 nodes, edges = lattice(square_unit_cell, (10, 10))
 
 psi = GraphState()
+v = VizClient()
 for node in nodes:
     psi.add_qubit(str(node), position=xyz(node[0], node[1]))
     psi.act_hadamard(str(node))
-    psi.update(0.1)
+    v.update(psi, 0.1)
 
 for edge in edges:
     psi.act_cz(str(edge[0]), str(edge[1]))
-psi.update(0.1)
+v.update(psi, 0.1)