Fix imports.

quantumlib · Oct 11, 2023 · 1d967cf · 1d967cf
1 parent a7e41d1
commit 1d967cf
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/qualtran/bloqs/chemistry/trotter.ipynb b/qualtran/bloqs/chemistry/trotter.ipynb
@@ -555,7 +555,8 @@
     "import cirq\n",
     "import cirq_ft\n",
     "\n",
-    "from qualtran.resource_counting import get_bloq_counts_graph, GraphvizCounts, SympySymbolAllocator\n",
+    "from qualtran.resource_counting import get_bloq_counts_graph, SympySymbolAllocator\n",
+    "from qualtran.drawing import GraphvizCounts\n",
     "from qualtran.bloqs.util_bloqs import Split, Join, Allocate, Free\n",
     "from qualtran.bloqs.basic_gates.rotation import RotationBloq\n",
     "from qualtran.cirq_interop import CirqGateAsBloq\n",
@@ -652,7 +653,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from qualtran.resource_counting import get_bloq_counts_graph, GraphvizCounts\n",
+    "from qualtran.resource_counting import get_bloq_counts_graph\n",
+    "from qualtran.drawing import GraphvizCounts\n",
     "from qualtran.bloqs.chemistry.trotter import KineticEnergy \n",
     "ke = KineticEnergy(4, 21)\n",
     "graph, sigma = get_bloq_counts_graph(ke, generalizer=generalize)\n",
@@ -665,7 +667,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from qualtran.resource_counting import get_bloq_counts_graph, GraphvizCounts\n",
+    "from qualtran.resource_counting import get_bloq_counts_graph\n",
+    "from qualtran.drawing import GraphvizCounts\n",
     "from qualtran.bloqs.chemistry.trotter import PotentialEnergy \n",
     "pe = PotentialEnergy(4, 21)\n",
     "graph, sigma = get_bloq_counts_graph(pe, generalizer=generalize)\n",