Generalization of Heterogeneous Multi-Robot Policies via Awareness and Communication of Capabilities