fix-ci (#1927)

metascroy · web-flow · commit 386e219c37a7 · 2025-03-19T16:39:16.000-07:00
diff --git a/torchao/experimental/ops/mps/test/test_lowbit.py b/torchao/experimental/ops/mps/test/test_lowbit.py
@@ -10,7 +10,8 @@
 import torch
 from parameterized import parameterized
 
-import torchao  # noqa: F401
+# Need to import to load the ops
+from torchao.experimental.quant_api import UIntxWeightOnlyLinearQuantizer  # noqa: F401
 
 try:
     for nbit in range(1, 8):
