diff --git a/tutorials/basic_loading_and_analysing.ipynb b/tutorials/basic_loading_and_analysing.ipynb index 33d93e1d..b25bf70f 100644 --- 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",