Merge pull request #256 from shunichironomura/add-warning-on-error

Add warning on encapsulation error
shunichironomura · Jun 29, 2024 · 67979e4 · 67979e4
2 parents 4855dac + 6d2ad3d
commit 67979e4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/capsula/encapsulator.py b/capsula/encapsulator.py
@@ -2,6 +2,7 @@
 
 import queue
 import threading
+import warnings
 from collections import OrderedDict
 from itertools import chain
 from typing import TYPE_CHECKING, Any, Tuple, Union
@@ -92,6 +93,7 @@ def encapsulate(self) -> Capsule:
             except Exception as e:  # noqa: PERF203
                 if capsule_item.abort_on_error:
                     raise
+                warnings.warn(f"Error occurred during encapsulation of {key}: {e}. Skipping.", stacklevel=3)
                 fails[key] = ExceptionInfo.from_exception(e)
         return Capsule(data, fails)