[ADD][custom_lint] Add fix pytlint error reimport w0404 and REF w0104 statement seems to have effect