fix relative paths

Author: aborgna-q

examples/python_optimization/python_optimization.ipynb

@@ -14,8 +14,9 @@
     "\n",
     "cwd = Path(\".\")\n",
     "notebook = next(cwd.rglob(\"python_optimization.ipynb\")).absolute()\n",
-    "jeff_path = notebook.parent.parent.parent / \"impl\" / \"py\" / \"src\"\n",
-    "sys.path.insert(0, str(jeff_path))\n",
+    "REPO_ROOT = notebook.parent.parent.parent\n",
+    "JEFF_PATH = REPO_ROOT / \"impl\" / \"py\" / \"src\"\n",
+    "sys.path.insert(0, str(JEFF_PATH))\n",
     "\n",
     "# avoid warning from capnp if \"PWD\" env variable is defined and different than the jupyter cwd\n",
     "import os\n",
@@ -140,9 +141,9 @@
     "import capnp\n",
     "\n",
     "capnp.remove_import_hook()\n",
-    "schema = capnp.load(str(jeff_path / \"jeff\" / \"data\" / \"jeff.capnp\"))\n",
+    "schema = capnp.load(str(JEFF_PATH / \"jeff\" / \"data\" / \"jeff.capnp\"))\n",
     "\n",
-    "with open(\"examples/catalyst_simple/catalyst_simple.jeff\", \"rb\") as f:\n",
+    "with open(REPO_ROOT / \"examples\" / \"catalyst_simple\" / \"catalyst_simple.jeff\", \"rb\") as f:\n",
     "    module = schema.Module.read(f)\n",
     "\n",
     "print(module)"
@@ -199,7 +200,7 @@
     "# the custom jeff bindings maintain the zero-copy nature of the capnp bindings when used for reading\n",
     "import jeff\n",
     "\n",
-    "module = jeff.load_module(\"examples/catalyst_simple/catalyst_simple.jeff\")\n",
+    "module = jeff.load_module(REPO_ROOT / \"examples\" / \"catalyst_simple\" / \"catalyst_simple.jeff\")\n",
     "print(module)"
    ]
   },
@@ -299,7 +300,7 @@
    "source": [
     "import jeff\n",
     "\n",
-    "module = jeff.load_module(\"examples/python_optimization/python_optimization.jeff\")\n",
+    "module = jeff.load_module(REPO_ROOT / \"examples\" / \"python_optimization\" / \"python_optimization.jeff\")\n",
     "print(module)"
    ]
   },
@@ -337,19 +338,19 @@
       "jeff v0\n",
       "\n",
       "[entry] func @main(int1) -> ():\n",
-      "  in : %4397829072:int1\n",
-      "    %4402776464:qubit = qubit.alloc \n",
-      "    %4402736208:qubit = qubit.gate %4402776464:qubit (h)\n",
-      "    %4402834256:int1 = int.not %4397829072:int1\n",
-      "    %4404055888:qubit = scf.switch %4402834256:int1, %4402736208:qubit \n",
+      "  in : %4364461904:int1\n",
+      "    %4364464112:qubit = qubit.alloc \n",
+      "    %4364463056:qubit = qubit.gate %4364464112:qubit (h)\n",
+      "    %4364465648:int1 = int.not %4364461904:int1\n",
+      "    %4365008976:qubit = scf.switch %4364465648:int1, %4364463056:qubit \n",
       "      case 0:\n",
-      "        in : %4404309008:qubit\n",
-      "          %4404304208:qubit = qubit.gate %4404309008:qubit (h)\n",
-      "        out: %4404304208:qubit\n",
+      "        in : %4364466992:qubit\n",
+      "          %4364467088:qubit = qubit.gate %4364466992:qubit (h)\n",
+      "        out: %4364467088:qubit\n",
       "      default:\n",
-      "        in : %4404302800:qubit\n",
-      "        out: %4404302800:qubit\n",
-      "    qubit.free %4404055888:qubit\n",
+      "        in : %4364467184:qubit\n",
+      "        out: %4364467184:qubit\n",
+      "    qubit.free %4365008976:qubit\n",
       "  out:\n",
       "\n"
      ]

impl/py/tests/test_examples.py

@@ -15,5 +15,6 @@ def test_example_notebooks(nb_regression, notebook: Path):
     nb_regression.diff_ignore += (
         "/metadata/language_info/version",
         "/cells/*/outputs/*/data/image/png",
+        "/cells/*/execution_count",
     )
     nb_regression.check(notebook)