fail to import torchao.quantization.pt2e

[shiyang-weng]

import torch,torchao
import torchao.quantization.pt2e

There is following issue:

Traceback (most recent call last):
File "", line 1, in
import torchao.quantization.pt2e
File "/home/wengshiy/miniforge3/envs/aotest_without_t/lib/python3.14/site-packages/torchao/quantization/pt2e/init.py", line 94, in
ObserverOrFakeQuantize.module = "torchao.quantization.pt2e"
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'typing.Union' object has no attribute 'module' and no dict for setting new attributes. Did you mean: 'reduce'?

My Env:
conda install python==3.14
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
pip3 install --pre torchao --index-url https://download.pytorch.org/whl/nightly/cpu

Could you help check it

[shiyang-weng]

I found that this issue only occurs in Python 3.14. But I need to test the performance on Python 3.14t because of performance issue pytorch/pytorch#170743

[danielvegamyhre]

I don't think we have official support for python 3.14, @jerryzh168 can correct me if i'm wrong

[jerryzh168]

yeah we haven't tested python 3.14, please feel free to submit a PR to fix if you have any ideas @shiyang-weng