CLN/TST: remove unused imports and correct the mpl.use function usage