chore: Fixed bug in the neuronpedia_integration cell in notebook (#402)

* corrected basic loading tutorial * fix formatting --------- Co-authored-by: David Chanin <[email protected]>
jbloomAus · Jan 12, 2025 · 4935b83 · 4935b83
1 parent c91763b
commit 4935b83
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/tutorials/basic_loading_and_analysing.ipynb b/tutorials/basic_loading_and_analysing.ipynb
@@ -456,13 +456,7 @@
     "from sae_lens.analysis.neuronpedia_integration import get_neuronpedia_quick_list\n",
     "\n",
     "# this function should open\n",
-    "neuronpedia_quick_list = get_neuronpedia_quick_list(\n",
-    "    test_feature_idx_gpt,\n",
-    "    layer=sae.cfg.hook_layer,\n",
-    "    model=\"gpt2-small\",\n",
-    "    dataset=\"res-jb\",\n",
-    "    name=\"A quick list we made\",\n",
-    ")\n",
+    "neuronpedia_quick_list = get_neuronpedia_quick_list(sae, test_feature_idx_gpt)\n",
     "\n",
     "if COLAB:\n",
     "    # If you're on colab, click the link below\n",