From 116b3fc5b8eb598e69bbc2c582e0ac0b966e00ea Mon Sep 17 00:00:00 2001
From: Hsin-Fang Chiang <hchiang@lsst.org>
Date: Tue, 25 Jun 2024 15:42:51 -0700
Subject: [PATCH] Check the count with the wanted instrument

---
 prompt-processing/groups.ipynb | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/prompt-processing/groups.ipynb b/prompt-processing/groups.ipynb
index d60ef72..dd16129 100644
--- a/prompt-processing/groups.ipynb
+++ b/prompt-processing/groups.ipynb
@@ -838,6 +838,17 @@
     "status_code = get_status_code_from_loki_fanout(date)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "1508e52e-b567-4092-805b-72111314d609",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "status_code = status_code[status_code[\"instrument\"] == instrument]\n",
+    "logger.debug(f\"{len(status_code)} are from {instrument}\")"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -845,7 +856,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "status_groups = status_code[status_code[\"instrument\"] == instrument].set_index([\"group\", \"detector\"]).groupby(\"code\").groups"
+    "status_groups = status_code.set_index([\"group\", \"detector\"]).groupby(\"code\").groups"
    ]
   },
   {