Skip to content

Commit 8f61ddc

Browse files
committed
use .call method for forward call
1 parent 3666e24 commit 8f61ddc

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

bioimageio/core/model_adapters/_tensorflow_model_adapter.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -200,14 +200,9 @@ def _forward_keras( # pyright: ignore[reportUnknownParameterType]
200200
None if ipt is None else tf.convert_to_tensor(ipt) for ipt in input_tensors
201201
]
202202

203-
try:
204-
result = ( # pyright: ignore[reportUnknownVariableType]
205-
self._network.forward(*tf_tensor)
206-
)
207-
except AttributeError:
208-
result = ( # pyright: ignore[reportUnknownVariableType]
209-
self._network.predict(*tf_tensor)
210-
)
203+
result = self._network.call( # pyright: ignore[reportUnknownVariableType]
204+
*tf_tensor
205+
)
211206

212207
if not isinstance(result, (tuple, list)):
213208
result = [result] # pyright: ignore[reportUnknownVariableType]

0 commit comments

Comments
 (0)