ENH: Cleaner mlab imports

@mwaskom what should we do about the `traitsui` versus `traits.ui` imports? Do we still need `try/except` for those?