Fix example 6 (#572)

This commit is contained in:
Simon Alibert
2024-12-11 10:32:18 +01:00
committed by GitHub
parent 44f9b21e74
commit 4c41f6fcc6
4 changed files with 10 additions and 9 deletions

View File

@@ -28,7 +28,7 @@ def safe_stop_image_writer(func):
try:
return func(*args, **kwargs)
except Exception as e:
dataset = kwargs.get("dataset", None)
dataset = kwargs.get("dataset")
image_writer = getattr(dataset, "image_writer", None) if dataset else None
if image_writer is not None:
print("Waiting for image writer to terminate...")