Skip to content

Commit 0417d71

Browse files
authored
Add nice error message when using wrong Slicer key (#1309)
1 parent e57a213 commit 0417d71

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/torchio/datasets/slicer.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,11 @@ class Slicer(Subject):
5050
"""
5151

5252
def __init__(self, name='MRHead'):
53-
filenames, url_files = URLS_DICT[name]
53+
try:
54+
filenames, url_files = URLS_DICT[name]
55+
except KeyError as e:
56+
message = f'Invalid name "{name}". Valid names are: {", ".join(URLS_DICT)}'
57+
raise ValueError(message) from e
5458
for filename, url_file in zip(filenames, url_files):
5559
filename = filename.replace('-', '_')
5660
url = urllib.parse.urljoin(SLICER_URL, url_file)

0 commit comments

Comments
 (0)