Там можно настраивать отдельные параметры. В данном случае tf ни при чем кмк, он подчеркивает неиспользованное. Это можно выключить, вот пример https://stackoverflow.com/questions/53357649/detect-unused-imports-in-visual-studio-code-for-python-3 но нужно именно ту опцию в документации найти, которая выключает unused import. Впрочем я использую, действительно, flake вместо pylint, так что это может быть нерелевантно.