Skip to content

Commit d3a2f4f

Browse files
committed
Merge remote-tracking branch 'Nic/8134-add-realtime-infer' into 8134-add-realtime-infer
2 parents 02a84b9 + de52a91 commit d3a2f4f

File tree

2 files changed

+13
-5
lines changed

2 files changed

+13
-5
lines changed

monai/bundle/workflows.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -398,9 +398,13 @@ def _run_expr(self, id: str, **kwargs: dict) -> list[Any]:
398398
ret = []
399399
if id in self.parser:
400400
# suppose all the expressions are in a list, run and reset the expressions
401-
for i in range(len(self.parser[id])):
402-
sub_id = f"{id}{ID_SEP_KEY}{i}"
403-
ret.append(self.parser.get_parsed_content(sub_id, **kwargs))
401+
if isinstance(self.parser[id], list):
402+
for i in range(len(self.parser[id])):
403+
sub_id = f"{id}{ID_SEP_KEY}{i}"
404+
ret.append(self.parser.get_parsed_content(sub_id, **kwargs))
405+
self.parser.ref_resolver.remove_resolved_content(sub_id)
406+
else:
407+
ret.append(self.parser.get_parsed_content(id, **kwargs))
404408
self.parser.ref_resolver.remove_resolved_content(sub_id)
405409
return ret
406410

tests/test_module_list.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,13 +58,17 @@ def test_transform_api(self):
5858
continue
5959
with self.subTest(n=n):
6060
basename = n[:-1] # Transformd basename is Transform
61+
62+
# remove aliases to check, do this before the assert below so that a failed assert does skip this
63+
for postfix in ("D", "d", "Dict"):
64+
remained.remove(f"{basename}{postfix}")
65+
6166
for docname in (f"{basename}", f"{basename}d"):
6267
if docname in to_exclude_docs:
6368
continue
6469
if (contents is not None) and f"`{docname}`" not in f"{contents}":
6570
self.assertTrue(False, f"please add `{docname}` to docs/source/transforms.rst")
66-
for postfix in ("D", "d", "Dict"):
67-
remained.remove(f"{basename}{postfix}")
71+
6872
self.assertFalse(remained)
6973

7074

0 commit comments

Comments
 (0)