ENH: Use int shape

This suppresses `numpy` warnings.