Congrats on the new addition to the family!
I prefer a wire cage. The best type to get would be one with an adjustable interior wall for training proposes.
Midwest Life Stages Folding Double Door Dog Crate 36" 1636DD#
I have some helpful guide lines to potty training that you may want to lool at...
Dawn's Yorkies / Yorkie Puppies / For Sale, Ohio