Revert "Remove unused import"

This reverts commit 046e54bc46f02c1aff72506eb0e470f55b67a180.
Have pylint ignore the import
Merge pull request #10557 from s0undt3ch/hotfix/lint

Revert "Remove unused import"