Small fixes in docs

Author: daniel-unyi-42

docs/notebooks/segger_tutorial.ipynb

@@ -506,17 +506,11 @@
     "    min_transcripts=5,\n",
     "    cell_id_col='segger_cell_id',\n",
     "    use_cc=False,\n",
-    "    knn_method='cuda',\n",
+    "    knn_method='kd_tree',\n",
     "    verbose=True,\n",
     ")\n"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "3aa5002c",
-   "metadata": {},
-   "source": []
-  },
   {
    "cell_type": "markdown",
    "id": "0a823035",

src/segger/models/segger_model.py

@@ -133,7 +133,7 @@ def decode(
         Decode the node embeddings to predict edge values.
 
         Args:
-            z (dict[str, Tensor]): Node embeddings for each node type.
+            z_dict (dict[str, Tensor]): Node embeddings for each node type.
             edge_index (EdgeIndex): Edge label indices.
 
         Returns: