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.
1 parent 8a06149 commit bacfd76Copy full SHA for bacfd76
bioimageio/core/backends/pytorch_backend.py
@@ -46,7 +46,7 @@ def __init__(
46
self._primary_device = devices[0]
47
48
def _forward_impl(
49
- self, input_arrays: Sequence[NDArray[Any] | None]
+ self, input_arrays: Sequence[Optional[NDArray[Any]]]
50
) -> List[Optional[NDArray[Any]]]:
51
tensors = [
52
None if a is None else torch.from_numpy(a).to(self._primary_device)
0 commit comments