could you do this on `stream` before partitioning?