up

Author: metascroy

.lintrunner.toml

@@ -392,10 +392,7 @@ command = [
   "lintrunner_adapters",
   "run",
   "grep_linter",
-  "--pattern=\\bfrom torch\\.ao\\.quantization\\.quantizer(?:\\.[A-Za-z0-9_]+)*\\b",
-  "--pattern=\\bfrom torch\\.ao\\.quantization\\.observer(?:\\.[A-Za-z0-9_]+)*\\b",
-  "--pattern=\\bfrom torch\\.ao\\.quantization\\.quantize_pt2e(?:\\.[A-Za-z0-9_]+)*\\b",
-  "--pattern=\\bfrom torch\\.ao\\.quantization\\.pt2e(?:\\.[A-Za-z0-9_]+)*\\b",
+  "--pattern=\\bfrom torch\\.ao\\.quantization\\.(?:quantizer|observer|quantize_pt2e|pt2e)(?:\\.[A-Za-z0-9_]+)*\\b",
   "--linter-name=TorchAOImport",
   "--error-name=Prohibited torch.ao.quantization import",
   """--error-description=\