Oh, right, the output dtype is not at all in the code.