Skip to content

Commit 8d4ed09

Browse files
committed
FIX(register): handle slice of pyramid levels
1 parent d0cfc42 commit 8d4ed09

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

nitorch/tools/registration/pairwise_pyramid.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,8 @@ def pyramid_levels(vxs, shapes, levels, **opt):
6969
for level in levels:
7070
if isinstance(level, int):
7171
select_levels.append(level)
72+
elif isinstance(level, slice):
73+
select_levels.extend(list(range(max_level))[level])
7274
else:
7375
select_levels.extend(list(level))
7476

0 commit comments

Comments
 (0)