Skip to content

Commit 82bb622

Browse files
committed
fix: warn about error that is not thrown
1 parent cd4236d commit 82bb622

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

bioimageio/core/model_adapters/_tensorflow_model_adapter.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,10 +106,10 @@ def _get_network( # pyright: ignore[reportUnknownParameterType]
106106
except Exception as e:
107107
try:
108108
return tf.keras.layers.TFSMLayer(
109-
weight_file, trainable=False, call_endpoint="serve_default"
109+
weight_file, trainable=False, call_endpoint="serving_default"
110110
) # pyright: ignore[reportUnknownVariableType]
111-
except Exception:
112-
logger.warning(f"error with `call_endpiont='serve': {e}")
111+
except Exception as ee:
112+
logger.warning(f"error with `call_endpiont='serving_default': {ee}")
113113
raise e
114114
else:
115115
# NOTE in tf1 the model needs to be loaded inside of the session, so we cannot preload the model

0 commit comments

Comments
 (0)