[minor] removed unnecessary arg from wandb zanj model loading

understanding-search · Oct 5, 2023 · 56947c6 · 56947c6
1 parent 9330d9a
commit 56947c6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/maze_transformer/utils/get_wandb_models.py b/maze_transformer/utils/get_wandb_models.py
@@ -200,7 +200,6 @@ def load_wandb_zanj(
     project: str = "aisc-search/alex",
     checkpoint: int | None = None,
     output_path: str | Path = "./downloaded_models",
-    model: bool = True,
 ) -> ZanjHookedTransformer:
     output_path = Path(output_path)
     api: wandb.Api = wandb.Api()

diff --git a/notebooks/get_wandb_models.ipynb b/notebooks/get_wandb_models.ipynb
@@ -41,11 +41,10 @@
    "outputs": [],
    "source": [
     "# load_wandb_zanj(\n",
-    "#     run_id=\"1n570yl5\",\n",
+    "# \trun_id=\"1n570yl5\",\n",
     "# \tproject=\"aisc-search/understanding-search\",\n",
-    "# \tcheckpoint=None,\n",
+    "# \tcheckpoint=None, # `None` for latest, or given checkpoint iter #\n",
     "# \toutput_path=\"./downloaded_models\",\n",
-    "# \tmodel=True,\n",
     "# )\n"
    ]
   }