Skip to content

Commit e3e2a53

Browse files
committed
improve error message
1 parent d01133c commit e3e2a53

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

bioimageio/core/model_adapters/_model_adapter.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import traceback
12
import warnings
23
from abc import ABC, abstractmethod
34
from typing import List, Optional, Sequence, Tuple, Union, final
@@ -76,7 +77,7 @@ def create(
7677
devices=devices,
7778
)
7879
except Exception as e:
79-
errors.append(f"{wf}: {e}")
80+
errors.append(f"{wf}: {e}\n{traceback.format_stack()}")
8081
elif (
8182
wf == "tensorflow_saved_model_bundle"
8283
and weights.tensorflow_saved_model_bundle is not None
@@ -88,7 +89,7 @@ def create(
8889
model_description=model_description, devices=devices
8990
)
9091
except Exception as e:
91-
errors.append(f"{wf}: {e}")
92+
errors.append(f"{wf}: {e}\n{traceback.format_stack()}")
9293
elif wf == "onnx" and weights.onnx is not None:
9394
try:
9495
from ._onnx_model_adapter import ONNXModelAdapter
@@ -97,7 +98,7 @@ def create(
9798
model_description=model_description, devices=devices
9899
)
99100
except Exception as e:
100-
errors.append(f"{wf}: {e}")
101+
errors.append(f"{wf}: {e}\n{traceback.format_stack()}")
101102
elif wf == "torchscript" and weights.torchscript is not None:
102103
try:
103104
from ._torchscript_model_adapter import TorchscriptModelAdapter
@@ -106,7 +107,7 @@ def create(
106107
model_description=model_description, devices=devices
107108
)
108109
except Exception as e:
109-
errors.append(f"{wf}: {e}")
110+
errors.append(f"{wf}: {e}\n{traceback.format_stack()}")
110111
elif wf == "keras_hdf5" and weights.keras_hdf5 is not None:
111112
# keras can either be installed as a separate package or used as part of tensorflow
112113
# we try to first import the keras model adapter using the separate package and,
@@ -124,7 +125,7 @@ def create(
124125
model_description=model_description, devices=devices
125126
)
126127
except Exception as e:
127-
errors.append(f"{wf}: {e}")
128+
errors.append(f"{wf}: {e}\n{traceback.format_stack()}")
128129

129130
assert errors
130131
error_list = "\n - ".join(errors)

0 commit comments

Comments
 (0)