Abstract
In some implementations, a cooling system for an electric machine may include an air inlet at a first end of the electric machine configured to provide an air flow through a housing of the electric machine. The cooling system may include one or more baffles removably coupled to a stator of the electric machine, the one or more baffles positioned proximate to a second end of the electric machine, and the one or more baffles being configured to direct the air flow toward one or more windings of the electric machine at the second end. The cooling system may include one or more openings in the housing of the electric machine proximate to the second end of the electric machine configured to enable the air flow to pass into or out of the housing.