From 9a5e4fa38ea88ec6e7e4e1fa4625a9660ae6fa90 Mon Sep 17 00:00:00 2001 From: Michiel Cottaar <MichielCottaar@gmail.com> Date: Fri, 23 Nov 2018 16:06:33 +0000 Subject: [PATCH] BUG: also allow job-ids to be integers --- fsl/utils/fslsub.py | 2 ++ tests/test_fslsub.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/fsl/utils/fslsub.py b/fsl/utils/fslsub.py index 5957ba6f3..a770cc623 100644 --- a/fsl/utils/fslsub.py +++ b/fsl/utils/fslsub.py @@ -225,6 +225,8 @@ def _flatten_job_ids(job_ids): """Unpack the (nested) job-ids in a single set""" if isinstance(job_ids, str): return {job_ids} + elif isinstance(job_ids, int): + return {str(job_ids)} else: res = set() for job_id in job_ids: diff --git a/tests/test_fslsub.py b/tests/test_fslsub.py index 45b4d9e0e..878e9b32c 100644 --- a/tests/test_fslsub.py +++ b/tests/test_fslsub.py @@ -9,7 +9,7 @@ from fsl.utils import fslsub def test_flatten_jobids(): - job_ids = ('12', '27', '35', '41', '721') + job_ids = ('12', '27', '35', '41', 721) res = '12,27,35,41,721' assert fslsub._flatten_job_ids(job_ids) == res -- GitLab