added missing import for TRACEMap implementation

The method ``_get_norm`` calls ``LogNorm`` but didn't import it.