Skip to content

Commit 448081c

Browse files
committed
chore: raise exception in notebook utils if thread has error
1 parent a93fbc7 commit 448081c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/sagemaker/jumpstart/notebook_utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -409,6 +409,9 @@ def evaluate_model(model_manifest: JumpStartModelHeader) -> Optional[Tuple[str,
409409
futures.append(executor.submit(evaluate_model, header))
410410

411411
for future in as_completed(futures):
412+
error = future.exception()
413+
if error:
414+
raise error
412415
result = future.result()
413416
if result:
414417
yield result

0 commit comments

Comments
 (0)