Fixed the HQQ import skip (#262)

pytorch · May 21, 2024 · 5e28109 · 5e28109
1 parent 3c9bc20
commit 5e28109
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/test/hqq/test_triton_mm.py b/test/hqq/test_triton_mm.py
@@ -5,12 +5,9 @@
     "triton", minversion="3.0.0", reason="Triton > 3.0.0 required to run this test"
 )
 hqq = pytest.importorskip("hqq", reason="hqq required to run this test")
-HQQLinear = pytest.importorskip(
-    "hqq.core.quantize.HQQLinear", reason="HQQLinear required to run this test"
-)
-BaseQuantizeConfig = pytest.importorskip(
-    "hqq.core.quantize.BaseQuantizeConfig", reason="HQQLinear required to run this test"
-)
+hqq_quantize = pytest.importorskip("hqq.core.quantize", reason="hqq required to run this test")
+HQQLinear = hqq_quantize.HQQLinear
+BaseQuantizeConfig = hqq_quantize.BaseQuantizeConfig
 
 import itertools