removed a TODO
remove TODO: einsum