Abstract: A cooling assembly is disclosed for use with an electric machine having bearings, a shaft rotatably supported by the bearings, a rotor connected to the shaft, and a stator annularly surrounding the rotor. The cooling assembly may have an end cap configured to engage an end of the rotor and the shaft to thereby axially locate the rotor relative to the shaft. The end cap may include an annular groove formed therein to receive a flow of cooling oil. The cooling assembly may further have an annular baffle axially connected to an end of the end cap and extending radially inward at least partially across an opening of the annular groove, and a slinger axially connected to the annular baffle opposite the end cap. The slinger may include a recess formed therein that is configured to trap the cooling oil after it flows through the annular baffle.