From 0bbf142b43ddd3e1dbf5028e2bf9eeae7b48d502 Mon Sep 17 00:00:00 2001
From: ahmedk <karim.ahmed@xfel.eu>
Date: Tue, 13 Aug 2024 16:39:55 +0200
Subject: [PATCH] feat: use exit_notebook with pnccd

---
 notebooks/pnCCD/Correct_pnCCD_NBC.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb b/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb
index 124216bbf..b19904f03 100644
--- a/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb
+++ b/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb
@@ -117,6 +117,7 @@
     "\n",
     "from cal_tools.tools import (\n",
     "    calcat_creation_time,\n",
+    "    exit_notebook,\n",
     "    write_constants_fragment,\n",
     ")\n",
     "from cal_tools.step_timing import StepTimer"
@@ -784,8 +785,7 @@
     "    step_timer.done_step(f'Storing data.')\n",
     "# Exit and raise warning if there are no data to correct for all sequences.\n",
     "if empty_seq == len(seq_files):\n",
-    "    warning(\"No valid trains for RAW data to correct.\")\n",
-    "    sys.exit(0)"
+    "    exit_notebook(\"No valid trains for RAW data to correct.\")"
    ]
   },
   {
-- 
GitLab