From 550730a2d6f67a7c60fdf81dbe03afff5be81a45 Mon Sep 17 00:00:00 2001 From: andrewor14 Date: Wed, 26 Nov 2025 12:23:26 -0800 Subject: [PATCH] Unbreak BC, add back PerRow, PerTensor imports **Summary:** OSS users like sglang are still importing these from `torchao.quantization.observer`. Here we quickly unbreak BC since that was not the intention of https://github.com/pytorch/ao/pull/3370. **Test Plan:** CI --- torchao/quantization/observer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchao/quantization/observer.py b/torchao/quantization/observer.py index 0e18770ae5..0aa9325a3e 100644 --- a/torchao/quantization/observer.py +++ b/torchao/quantization/observer.py @@ -12,7 +12,7 @@ from torchao.quantization.quant_primitives import _fake_quantize_affine -from .granularity import Granularity +from .granularity import Granularity, PerRow, PerTensor # noqa: F401 from .quant_primitives import ( MappingType, ZeroPointDomain,