Skip to content

Commit b966b34

Browse files
committed
fix keras network call
1 parent fe45db8 commit b966b34

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bioimageio/core/model_adapters/_keras_model_adapter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def __init__(
7979
def forward(self, *input_tensors: Optional[Tensor]) -> List[Optional[Tensor]]:
8080
_result: Union[Sequence[NDArray[Any]], NDArray[Any]]
8181
_result = self._network.predict( # pyright: ignore[reportUnknownVariableType]
82-
*input_tensors
82+
*[None if t is None else t.data.data for t in input_tensors]
8383
)
8484
if isinstance(_result, (tuple, list)):
8585
result: Sequence[NDArray[Any]] = _result

0 commit comments

Comments
 (0)