diff --git a/notebooks/LPD/LPDChar_Darks_NBC.ipynb b/notebooks/LPD/LPDChar_Darks_NBC.ipynb index f7924cdb20c4a36821132fa5ad7856f20206bf85..de76f651d6b3f713b3e96102a8dd793eb850a3d9 100644 --- a/notebooks/LPD/LPDChar_Darks_NBC.ipynb +++ b/notebooks/LPD/LPDChar_Darks_NBC.ipynb @@ -189,8 +189,12 @@ " data = data[skip_first_ntrains : skip_first_ntrains + ntrains]\n", " cell_ids = run[det_source, 'image.cellId'].drop_empty_trains()\n", " cell_ids = cell_ids[skip_first_ntrains : skip_first_ntrains + ntrains]\n", - " \n", - " if len(data.train_ids) < min_trains:\n", + "\n", + " # If there is no data available, return and expect this\n", + " # module to be skipped later.\n", + " if len(data.train_ids) == 0:\n", + " return None, None, None, None, None, None, None, None\n", + " elif len(data.train_ids) < min_trains:\n", " raise Exception(f\"Run {run_path} only contains {len(data.train_ids)} trains, but {min_trains} required\")\n", "\n", " im = data.ndarray()\n",