fusedbloxxer

Functions that Return Unions are not Friendly with Linters

1

The [einx.rearrange](https://github.com/fferflo/einx/blob/b7a90b413f74f7d215d2deedd25a760057e4ee2d/einx/op/rearrange.py#L92) function returns a `Union[einx.Tensor, Tuple[einx.Tensor, ...]]` and this is problematic when assigning the result into a variable with an `einx.Tensor` typehint, because the function might be returning a...