1 files changed, 12 insertions, 10 deletions

diff --git a/analysis/masks/visualize.ipynb b/analysis/masks/visualize.ipynb
index 08a7ee3..5ec4a52 100644
--- a/analysis/masks/visualize.ipynb
+++ b/analysis/masks/visualize.ipynb
@@ -15,7 +15,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 74,
+   "execution_count": 1,
    "id": "b5ca2f23-cdd2-4475-84a6-fd7f273f1445",
    "metadata": {},
    "outputs": [],
@@ -50,14 +50,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 75,
+   "execution_count": 10,
    "id": "88c64f99-acf5-48df-bca2-a53624552bf8",
    "metadata": {},
    "outputs": [],
    "source": [
     "def load_collection(collection_path):\n",
     "    collection = []\n",
-    "    run_files = [file for file in os.listdir(run_path) if file.endswith(\".txt\")]\n",
+    "    run_files = [file for file in os.listdir(collection_path) if file.endswith(\".txt\")]\n",
     "    run_files = sorted(run_files, key = lambda x: int(x[:-4].split(\"_\")[-1]))\n",
     "    sizes = set()\n",
     "    for file in run_files:\n",
@@ -68,8 +68,10 @@
     "def load_collections(card, protocol):\n",
     "    path = f\"results/{card}/testdn/\"\n",
     "    collections = []\n",
+    "    collection_dirs = list(os.listdir(path))\n",
+    "    collection_dirs.sort()\n",
     "    counter = 0\n",
-    "    for collection_dir in os.listdir(path):\n",
+    "    for collection_dir in collection_dirs:\n",
     "        if not collection_dir.startswith(protocol):\n",
     "            continue\n",
     "        collection_path = os.path.join(path, collection_dir)\n",
@@ -97,7 +99,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 83,
+   "execution_count": 11,
    "id": "6e5c5eb1-00d6-4890-a09c-996a8fd7f839",
    "metadata": {},
    "outputs": [
@@ -105,8 +107,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "(0) Loading 1000x100 masks from results/N9/testdn/ecdsa_rk_18_15_52_Jun_11_2025_60889d3ffb92583d96d09cb818a36c67680893b5\n",
-      "(1) Loading 5x1000 masks from results/N9/testdn/ecdsa_rk_12_23_17_Jun_12_2025_60889d3ffb92583d96d09cb818a36c67680893b5\n"
+      "(0) Loading collection with 5x1000 masks from results/N9/testdn/ecdsa_rk_12_23_17_Jun_12_2025_60889d3ffb92583d96d09cb818a36c67680893b5\n",
+      "(1) Loading collection with 1000x100 masks from results/N9/testdn/ecdsa_rk_18_15_52_Jun_11_2025_60889d3ffb92583d96d09cb818a36c67680893b5\n"
      ]
     }
    ],
@@ -129,7 +131,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 86,
+   "execution_count": 12,
    "id": "d55ed262-3f5c-4f65-953e-cbf82987a3b5",
    "metadata": {},
    "outputs": [
@@ -145,7 +147,7 @@
     }
    ],
    "source": [
-    "masks_multiple = collections[0][:20]\n",
+    "masks_multiple = collections[1][:20]\n",
     "\n",
     "num = len(masks_multiple)\n",
     "ncols = 4\n",
@@ -174,7 +176,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 88,
+   "execution_count": 14,
    "id": "28333ccd-4a1b-47f1-87ef-e735088a870c",
    "metadata": {},
    "outputs": [