Commit 9a5e4fa3 authored by Michiel Cottaar's avatar Michiel Cottaar Committed by Paul McCarthy
Browse files

BUG: also allow job-ids to be integers

parent 3287a371
...@@ -225,6 +225,8 @@ def _flatten_job_ids(job_ids): ...@@ -225,6 +225,8 @@ def _flatten_job_ids(job_ids):
"""Unpack the (nested) job-ids in a single set""" """Unpack the (nested) job-ids in a single set"""
if isinstance(job_ids, str): if isinstance(job_ids, str):
return {job_ids} return {job_ids}
elif isinstance(job_ids, int):
return {str(job_ids)}
else: else:
res = set() res = set()
for job_id in job_ids: for job_id in job_ids:
......
...@@ -9,7 +9,7 @@ from fsl.utils import fslsub ...@@ -9,7 +9,7 @@ from fsl.utils import fslsub
def test_flatten_jobids(): def test_flatten_jobids():
job_ids = ('12', '27', '35', '41', '721') job_ids = ('12', '27', '35', '41', 721)
res = '12,27,35,41,721' res = '12,27,35,41,721'
assert fslsub._flatten_job_ids(job_ids) == res assert fslsub._flatten_job_ids(job_ids) == res
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment