(.venv) ➜ tmp python3 -c "from torchao.quantization import quantize_"
Traceback (most recent call last):
File "<string>", line 1, in <module>
from torchao.quantization import quantize_
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torchao/__init__.py", line 110, in <module>
from torchao.quantization import (
...<2 lines>...
)
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torchao/quantization/__init__.py", line 6, in <module>
from .autoquant import (
...<8 lines>...
)
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torchao/quantization/autoquant.py", line 11, in <module>
from torchao.dtypes import (
...<6 lines>...
)
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torchao/dtypes/__init__.py", line 1, in <module>
from . import affine_quantized_tensor_ops
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torchao/dtypes/affine_quantized_tensor_ops.py", line 11, in <module>
from torchao.dtypes.affine_quantized_tensor import (
AffineQuantizedTensor,
)
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torchao/dtypes/affine_quantized_tensor.py", line 17, in <module>
from torchao.quantization.quant_primitives import (
...<19 lines>...
)
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torchao/quantization/quant_primitives.py", line 206, in <module>
register_custom_op = _register_custom_op(quant_lib)
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torchao/utils.py", line 210, in _register_custom_op
from torch._inductor.decomposition import register_decomposition
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torch/_inductor/decomposition.py", line 14, in <module>
import torch.ao.quantization.fx._decomposed
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torch/ao/quantization/fx/__init__.py", line 3, in <module>
from .prepare import prepare
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torch/ao/quantization/fx/prepare.py", line 31, in <module>
from torch.ao.quantization.quantizer import (
...<6 lines>...
)
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torch/ao/quantization/quantizer/__init__.py", line 1, in <module>
from .quantizer import (
...<8 lines>...
)
File "/Users/ownia/codespace/tmp/.venv/lib/python3.14/site-packages/torch/ao/quantization/quantizer/quantizer.py", line 85, in <module>
EdgeOrNode.__module__ = "torch.ao.quantization.quantizer.quantizer"
^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'typing.Union' object has no attribute '__module__' and no __dict__ for setting new attributes. Did you mean: '__reduce__'?
