We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
.call
1 parent 3666e24 commit 8f61ddcCopy full SHA for 8f61ddc
bioimageio/core/model_adapters/_tensorflow_model_adapter.py
@@ -200,14 +200,9 @@ def _forward_keras( # pyright: ignore[reportUnknownParameterType]
200
None if ipt is None else tf.convert_to_tensor(ipt) for ipt in input_tensors
201
]
202
203
- try:
204
- result = ( # pyright: ignore[reportUnknownVariableType]
205
- self._network.forward(*tf_tensor)
206
- )
207
- except AttributeError:
208
209
- self._network.predict(*tf_tensor)
210
+ result = self._network.call( # pyright: ignore[reportUnknownVariableType]
+ *tf_tensor
+ )
211
212
if not isinstance(result, (tuple, list)):
213
result = [result] # pyright: ignore[reportUnknownVariableType]
0 commit comments