fix unittest

lyogavin · lyogavin · commit 7f9a65c43cbe · 2025-11-26T12:41:22.000-06:00
diff --git a/tests-unit/comfy_quant/test_quant_registry.py b/tests-unit/comfy_quant/test_quant_registry.py
@@ -1453,6 +1453,7 @@ def test_int8_addmm_with_out_quant(self):
             bias, input_q, weight_q.t()
         )
         
+        output_quant = QuantizedTensor.from_float(output_quant, "BlockWiseINT8Layout", block_size=block_size, is_weight=False)
         self.assertIsInstance(output_quant, QuantizedTensor, "Output should be QuantizedTensor when out_quant=True")
         self.assertEqual(output_quant.shape, (M, N))
         self.assertEqual(output_quant._layout_type, "BlockWiseINT8Layout")

Original file line number	Diff line number	Diff line change
`@@ -1453,6 +1453,7 @@ def test_int8_addmm_with_out_quant(self):`
`1453`	`1453`	`bias, input_q, weight_q.t()`
`1454`	`1454`	`)`
`1455`	`1455`
	`1456`	`+ output_quant = QuantizedTensor.from_float(output_quant, "BlockWiseINT8Layout", block_size=block_size, is_weight=False)`
`1456`	`1457`	`self.assertIsInstance(output_quant, QuantizedTensor, "Output should be QuantizedTensor when out_quant=True")`
`1457`	`1458`	`self.assertEqual(output_quant.shape, (M, N))`
`1458`	`1459`	`self.assertEqual(output_quant._layout_type, "BlockWiseINT8Layout")`