diff --git a/cyto_dl/datamodules/multidim_image.py b/cyto_dl/datamodules/multidim_image.py index 8bf1bb01..2612be92 100644 --- a/cyto_dl/datamodules/multidim_image.py +++ b/cyto_dl/datamodules/multidim_image.py @@ -101,7 +101,7 @@ def _get_timepoints(self, row, img): start = row.get(self.time_start_column, 0) stop = row.get(self.time_stop_column, -1) step = row.get(self.time_step_column, 1) - timepoints = range(start, stop + 1, step) if stop > 0 else range(img.dims.T) + timepoints = range(start, stop + 1, step) if stop >= 0 else range(img.dims.T) return list(timepoints) def get_per_file_args(self, df):