So I'm just now figuring out that when you create a custom faction for the AI to run that it doesn't "use" the traits you give the faction but instead uses units based on the race of the faction. So if I design a Krax raced faction that uses Great Hammers the AI won't know how to use those weapons on most of its builds, instead it will build basic units (as most of the AI Krax units are one-handed spear units). 

Is there an easy way to fix this without having to create a bunch of units via XML (or by playing a game as each custom faction)?

Thanks in advance!