Updated basic_example to use load_ops

MouseLand · Jun 17, 2024 · 018fe09 · 018fe09
1 parent 633ee92
commit 018fe09
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/docs/tutorials/basic_example.ipynb b/docs/tutorials/basic_example.ipynb
@@ -143,13 +143,17 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from pathlib import Path\n",
+    "\n",
     "import numpy as np\n",
     "import pandas as pd\n",
-    "from pathlib import Path\n",
+    "\n",
+    "from kilosort.io import load_ops\n",
+    "\n",
     "\n",
     "# outputs saved to results_dir\n",
     "results_dir = Path(settings['data_dir']).joinpath('kilosort4')\n",
-    "ops = np.load(results_dir / 'ops.npy', allow_pickle=True).item()\n",
+    "ops = load_ops(results_dir / 'ops.npy')\n",
     "camps = pd.read_csv(results_dir / 'cluster_Amplitude.tsv', sep='\\t')['Amplitude'].values\n",
     "contam_pct = pd.read_csv(results_dir / 'cluster_ContamPct.tsv', sep='\\t')['ContamPct'].values\n",
     "chan_map =  np.load(results_dir / 'channel_map.npy')\n",