diff --git a/docs/tutorials/basic_example.ipynb b/docs/tutorials/basic_example.ipynb index e7aace33..ceb7ba9a 100644 --- 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",